class RandomDelete:
def __init__(self, vectors, aug_p=0.3, stopwords=[]):
self.swap_aug = WordEmbsAug(model=vectors,
action='delete',
stopwords=stopwords,
aug_p=aug_p)
def __call__(self, text):
text_transformed = self.swap_aug.augment(text)
return text_transformed
class RandomDelete:
def __init__(self, vectors, aug_p=0.3):
self.swap_aug = WordEmbsAug(
model=vectors,
action='substitute',
stopwords=['<cls>', '<eos>', '<sep>', '<pad>'],
aug_p=aug_p)
def __call__(self, text):
text_transformed = self.swap_aug.augment(text)
return text_transformed
def __init__(self, vectors, aug_p=0.3):
self.swap_aug = WordEmbsAug(
model=vectors,
action='substitute',
stopwords=['<cls>', '<eos>', '<sep>', '<pad>'],
aug_p=aug_p)
def __init__(self, vectors, aug_p=0.3, stopwords=[]):
self.swap_aug = WordEmbsAug(model=vectors,
action='delete',
stopwords=stopwords,
aug_p=aug_p)