sparknlp.annotator.tf_ner_dl_graph_builder#

Classes#

class TFNerDLGraphBuilderModel[source]#

Abstract class for models that are fitted by estimators.

New in version 1.4.0.

class TFNerDLGraphBuilder[source]#

Abstract class for estimators that fit models to data.

New in version 1.3.0.

setHiddenUnitsNumber(value)[source]#

Sets the number of hidden units for AssertionDLApproach and MedicalNerApproach

Parameters:

valueint

Number of hidden units for AssertionDLApproach and MedicalNerApproach

getHiddenUnitsNumber()[source]#

Gets the number of hidden units for AssertionDLApproach and MedicalNerApproach.

setLabelColumn(value)[source]#

Sets the name of the column for data labels.

Parameters:

valuestr

Column for data labels

getLabelColumn()[source]#

Gets the name of the label column.

setInputCols(*value)[source]#

Sets column names of input annotations.

Parameters:

*valueList[str]

Input columns for the annotator

getInputCols()[source]#

Gets current column names of input annotations.

setGraphFolder(value)[source]#

Sets folder path that contain external graph files.

Parameters:

valuesrt

Folder path that contain external graph files.

getGraphFolder()[source]#

Gets the graph folder.

setGraphFile(value)[source]#

Sets the graph file name.

Parameters:

valuesrt

Greaph file name. If set to “auto”, then the graph builder will use the model specific default graph file name.

getGraphFile()[source]#

Gets the graph file name.