Skip to content
This repository was archived by the owner on Aug 9, 2023. It is now read-only.

Pre-release

Pre-release
Pre-release
Compare
Choose a tag to compare
@aCampello aCampello released this 31 Jan 10:37
513b292

This is the first (pre)-release of WellcomeML. Some features

ML

  • wellcomeml.ml.BertClassifier: sklearn-compatible Bert and Scibert classifier (i.e., with fit and predict)
  • wellcomeml.ml.BertVectorizer: sklearn-compatible Bert and Scibert vectorizer (i.e., with fit and transform)
  • wellcomeml.ml.SpacyNER: sklearn compatible Spacy named-entity recognition
  • wellcomeml.ml.SpacyClassifier: sklearn-compatible Spacy classifier

IO

  • wellcomeml.io.PolicyDocumentsDownloader: Utility class to download policy documents

Metrics

  • wellcomeml.metrics.ner_classification_report(): Method to generate pretty classification reports for named-entity recognition