nessie.models.tagging.flair_sequence_tagger
Module Contents
Classes
Helper class that provides a standard way to create an ABC using |
- class nessie.models.tagging.flair_sequence_tagger.FlairSequenceTagger(verbose: bool = True, max_epochs: int = 150, batch_size: int = 64)
Bases:
nessie.models.SequenceTagger
Helper class that provides a standard way to create an ABC using inheritance.
- TAG_TYPE = tag
- fit(self, X: nessie.types.RaggedStringArray, y: nessie.types.RaggedStringArray)
- has_dropout(self) bool
- label_encoder(self) sklearn.preprocessing.LabelEncoder
Returns a label encoder that can be used to map labels to ints and vice versa
- predict(self, X: nessie.types.RaggedStringArray) awkward.Array
- predict_proba(self, X: nessie.types.RaggedStringArray) awkward.Array
Returns a distribution over labels for each instance.
- Parameters
X – The token sequences to predict on
- Returns
A (num_sentences, num_tokens, num_labels) ragged array
- score(self, X: nessie.types.RaggedStringArray) awkward.Array
Returns the best score for each item
- use_dropout(self, is_activated: bool)