Skip to content

Initial public release

Choose a tag to compare

@jasonnance jasonnance released this 04 Sep 21:11
· 246 commits to master since this release

Initial public release includes the following models:

  • BERT (training, prediction, embeddings)
  • MT-DNN (training, prediction, embeddings)
  • Universal Sentence Encoder (embeddings)
  • fastText (training, prediction, embeddings)
  • pytorch_transformer Transformers (training, prediction, embeddings)

Additional features include:

  • Experiments (ClassificationExperiments only)
  • Data augmentation (Word2Vec, WordNet, and BERTMaskedLM)
  • Document windowing