Spaces:
Sleeping
Sleeping
File size: 676 Bytes
4a1df2e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
"""
Min Word Lenth
--------------------------
"""
from textattack.constraints import PreTransformationConstraint
class MinWordLength(PreTransformationConstraint):
"""A constraint that prevents modifications to words less than a certain
word character-length.
:param min_length: Minimum word character-length needed for changes to be made to a word.
"""
def __init__(self, min_length):
self.min_length = min_length
def _get_modifiable_indices(self, current_text):
idxs = []
for i, word in enumerate(current_text.words):
if len(word) >= self.min_length:
idxs.append(i)
return set(idxs)
|