sparknlp.annotator.param.classifier_encoder#

Classes#

class ClassifierEncoder[source]#

Components that take parameters. This also provides an internal param map to store parameter values attached to the instance.

New in version 1.3.0.

setMaxEpochs(epochs)[source]#

Sets maximum number of epochs to train, by default 30

Parameters:

epochsint

Maximum number of epochs to train

setLr(v)[source]#

Sets Learning Rate, by default 0.005

Parameters:

vfloat

Learning Rate

setBatchSize(v)[source]#

Sets batch size, by default 64.

Parameters:

vint

Batch size

setLabelColumn(value)[source]#

Sets name of column for data labels

Parameters:

valuestr

Column for data labels

setRandomSeed(seed)[source]#

Sets random seed for shuffling

Parameters:

seedint

Random seed for shuffling

setConfigProtoBytes(b)[source]#

Sets configProto from tensorflow, serialized into byte array.

Parameters:

bList[int]

ConfigProto from tensorflow, serialized into byte array