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

Releases: wellcometrust/WellcomeML

Pre-release

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

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