sparknlp.annotator.ner.ner_approach#

Contains base classes for NER Annotators.

Module Contents#

Classes#

NerApproach

Base class for Ner*Approach Annotators

class NerApproach[source]#

Base class for Ner*Approach Annotators

setLabelColumn(value)[source]#

Sets name of column for data labels.

Parameters:
valuestr

Column for data labels

setEntities(tags)[source]#

Sets entities to recognize.

Parameters:
tagsList[str]

List of entities

setMinEpochs(epochs)[source]#

Sets minimum number of epochs to train.

Parameters:
epochsint

Minimum number of epochs to train

setMaxEpochs(epochs)[source]#

Sets maximum number of epochs to train.

Parameters:
epochsint

Maximum number of epochs to train

setRandomSeed(seed)[source]#

Sets random seed for shuffling.

Parameters:
seedint

Random seed for shuffling

getLabelColumn()[source]#

Gets column for label per each token.

Returns:
str

Column with label per each token