org.tribuo.interop.tensorflow.sequence

Class TensorflowSequenceModel<T extends Output<T>>

java.lang.Object

org.tribuo.sequence.SequenceModel<T>

org.tribuo.interop.tensorflow.sequence.TensorflowSequenceModel<T>

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.provenance.Provenancable<ModelProvenance>, Closeable, Serializable, AutoCloseable

public class TensorflowSequenceModel<T extends Output<T>>
extends SequenceModel<T>
implements Closeable

A Tensorflow model which implements SequenceModel, suitable for use in sequential prediction tasks.

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected SequenceExampleTransformer<T>	exampleTransformer
protected String	initOp
protected SequenceOutputTransformer<T>	outputTransformer
protected String	predictOp

Fields inherited from class org.tribuo.sequence.SequenceModel

featureIDMap, name, outputIDMap, provenanceOutput

Method Summary

Modifier and Type	Method and Description
void	close() Close the session and graph if they exist.
Map<String,List<com.oracle.labs.mlrg.olcut.util.Pair<String,Double>>>	getTopFeatures(int i) Returns an empty map, as the top features are not well defined for most Tensorflow models.
List<Prediction<T>>	predict(SequenceExample<T> example) Uses the model to predict the output for a single example.

Methods inherited from class org.tribuo.sequence.SequenceModel

getFeatureIDMap, getName, getOutputIDInfo, getProvenance, predict, predict, setName, toMaxLabels, toString, validate

Methods inherited from class java.lang.Object

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

Field Detail

exampleTransformer

protected final SequenceExampleTransformer<T extends Output<T>> exampleTransformer

outputTransformer

protected final SequenceOutputTransformer<T extends Output<T>> outputTransformer

initOp

protected final String initOp

predictOp

protected final String predictOp

Method Detail

predict

public List<Prediction<T>> predict(SequenceExample<T> example)

Description copied from class: SequenceModel

Uses the model to predict the output for a single example.

Specified by:

predict in class SequenceModel<T extends Output<T>>

Parameters:

example - the example to predict.

Returns:

the result of the prediction.

getTopFeatures

public Map<String,List<com.oracle.labs.mlrg.olcut.util.Pair<String,Double>>> getTopFeatures(int i)

Returns an empty map, as the top features are not well defined for most Tensorflow models.

Specified by:

getTopFeatures in class SequenceModel<T extends Output<T>>

Parameters:

i - the number of features to return. If this value is less than 0, all features should be returned for each class, unless the model cannot score it's features.

Returns:

a map from string outputs to an ordered list of pairs of feature names and weights associated with that feature in the model

close

public void close()

Close the session and graph if they exist.

Specified by:

close in interface Closeable

Specified by:

close in interface AutoCloseable