Package com.gengoai.hermes.ml

Class TensorFlowSequenceLabeler

java.lang.Object

com.gengoai.apollo.ml.model.TensorFlowModel

com.gengoai.hermes.ml.TensorFlowSequenceLabeler

All Implemented Interfaces:

Model, Transform, Copyable<Transform>, SerializableFunction<HString,HString>, HStringMLModel, Serializable, Function<HString,HString>

Direct Known Subclasses:

ElmoSeq2SeqModel, NeuralNERModel

public abstract class TensorFlowSequenceLabeler
extends TensorFlowModel
implements HStringMLModel

Author:

David B. Bracewell

See Also:

Serialized Form

Field Summary

Fields inherited from class com.gengoai.apollo.ml.model.TensorFlowModel

inputs, modelFile, outputs, transformer

Constructor Summary

Constructors

Constructor	Description
TensorFlowSequenceLabeler(@NonNull Map<String,TFVarSpec> inputs, @NonNull LinkedHashMap<String,TFVarSpec> outputs, @NonNull SequenceValidator validator, @NonNull TagDecoder tagDecoder)

Method Summary

Modifier and Type	Method	Description
HString	apply(HString hString)
protected Observation	decodeNDArray(String name, NDArray ndArray)
Model	delegate()	Delegate model.
LabelType	getLabelType(@NonNull String name)
void	setVersion(String version)	Sets version.

Methods inherited from class com.gengoai.apollo.ml.model.TensorFlowModel

calculate_max_sequence_length, createTensors, createTransformer, estimate, getFitParameters, getInputs, getOutputs, load, processBatch, save, setEncoder, transform, transform

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface java.util.function.Function

andThen, compose

Methods inherited from interface com.gengoai.hermes.ml.HStringMLModel

estimate, estimate, getDataGenerator, getEvaluator, getFitParameters, getInputs, getOutput, getOutputs, getVersion, transform, transform, transform, transform

Methods inherited from interface com.gengoai.apollo.ml.model.Model

copy, fitAndTransform, getLabelType, save

Methods inherited from interface com.gengoai.function.SerializableFunction

andThen, asConsumer, compose

Constructor Detail

TensorFlowSequenceLabeler

public TensorFlowSequenceLabeler(@NonNull
                                 @NonNull Map<String,TFVarSpec> inputs,
                                 @NonNull
                                 @NonNull LinkedHashMap<String,TFVarSpec> outputs,
                                 @NonNull
                                 @NonNull SequenceValidator validator,
                                 @NonNull
                                 @NonNull TagDecoder tagDecoder)

Method Detail

apply

public HString apply(HString hString)

Specified by:

apply in interface Function<HString,HString>

decodeNDArray

protected Observation decodeNDArray(String name,
                                    NDArray ndArray)

Specified by:

decodeNDArray in class TensorFlowModel

delegate

public Model delegate()

Description copied from interface: HStringMLModel

Delegate model.

Specified by:

delegate in interface HStringMLModel

Returns:

the model

getLabelType

public LabelType getLabelType(@NonNull
                              @NonNull String name)

Specified by:

getLabelType in interface HStringMLModel

Specified by:

getLabelType in interface Model

setVersion

public void setVersion(String version)

Description copied from interface: HStringMLModel

Sets version.

Specified by:

setVersion in interface HStringMLModel

Parameters:

version - the version