idiomify / explore /explore_nlpaug.py
eubinecto's picture
[#1] refactoring: Alpha -> Seq2Seq. We rely on git tags for versioning models.
64a6414
import nlpaug.augmenter.word as naw
import nlpaug.augmenter.sentence as nas
import nltk
sent = "I am really happy with the new job and I mean that with sincere feeling"
def main():
nltk.download("omw-1.4")
# this seems legit! I could definitely use this to increase the accuracy of the model
# for a few idioms (possibly ten, ten very different but frequent idioms)
aug = naw.ContextualWordEmbsAug()
augmented = aug.augment(sent, n=10)
print(augmented)
if __name__ == '__main__':
main()