Package com.gengoai.hermes.ml.model
Class NeuralNERModel
- java.lang.Object
-
- com.gengoai.apollo.ml.model.TensorFlowModel
-
- com.gengoai.hermes.ml.TensorFlowSequenceLabeler
-
- com.gengoai.hermes.ml.model.NeuralNERModel
-
- All Implemented Interfaces:
Model
,Transform
,Copyable<Transform>
,SerializableFunction<HString,HString>
,HStringMLModel
,Serializable
,Function<HString,HString>
public class NeuralNERModel extends TensorFlowSequenceLabeler 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 NeuralNERModel()
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description protected int
calculate_max_sequence_length(DataSet batch)
HStringDataSetGenerator
getDataGenerator()
Gets data generator.String
getVersion()
Gets version.static void
main(String[] args)
-
Methods inherited from class com.gengoai.hermes.ml.TensorFlowSequenceLabeler
apply, decodeNDArray, delegate, getLabelType, setVersion
-
Methods inherited from class com.gengoai.apollo.ml.model.TensorFlowModel
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 com.gengoai.hermes.ml.HStringMLModel
delegate, estimate, estimate, getEvaluator, getFitParameters, getInputs, getLabelType, getOutput, getOutputs, setVersion, 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
-
-
-
-
Method Detail
-
calculate_max_sequence_length
protected int calculate_max_sequence_length(DataSet batch)
- Overrides:
calculate_max_sequence_length
in classTensorFlowModel
-
getDataGenerator
public HStringDataSetGenerator getDataGenerator()
Description copied from interface:HStringMLModel
Gets data generator.- Specified by:
getDataGenerator
in interfaceHStringMLModel
- Returns:
- the data generator
-
getVersion
public String getVersion()
Description copied from interface:HStringMLModel
Gets version.- Specified by:
getVersion
in interfaceHStringMLModel
- Returns:
- the version
-
-