Package | Description |
---|---|
org.tribuo.interop.tensorflow |
Provides an interface to TensorFlow, allowing the training of non-sequential models using any supported
Tribuo output type.
|
Modifier and Type | Method and Description |
---|---|
static TensorFlowUtil.TensorTuple |
TensorFlowUtil.TensorTuple.of(org.tensorflow.types.family.TType tensor)
Makes a TensorTuple out of this tensor.
|
Modifier and Type | Method and Description |
---|---|
static 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 Type | Method and Description |
---|---|
static 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.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.