Uses of Class
org.tribuo.interop.tensorflow.TensorFlowUtil.TensorTuple
Package
Description
Provides an interface to TensorFlow, allowing the training of non-sequential models using any supported
Tribuo output type.
-
Uses of TensorFlowUtil.TensorTuple in org.tribuo.interop.tensorflow
Modifier and TypeMethodDescriptionstatic TensorFlowUtil.TensorTuple
TensorFlowUtil.TensorTuple.of
(org.tensorflow.types.family.TType tensor) Makes a TensorTuple out of this tensor.Modifier and TypeMethodDescriptionstatic Map<String,
TensorFlowUtil.TensorTuple> TensorFlowUtil.extractMarshalledVariables
(org.tensorflow.Graph graph, org.tensorflow.Session session) Extracts a Map containing the name of each Tensorflow VariableV2 and the associated parameter array.Modifier and TypeMethodDescriptionstatic void
TensorFlowUtil.restoreMarshalledVariables
(org.tensorflow.Session session, Map<String, TensorFlowUtil.TensorTuple> tensorMap) Writes a map containing the name of each Tensorflow VariableV2 and the associated parameter array into the supplied session.