| 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 | Class and Description |
|---|---|
class |
DenseFeatureConverter
Converts a sparse example into a dense float vector, then wraps it in a
TFloat32. |
class |
ImageConverter
Image converter.
|
| Modifier and Type | Field and Description |
|---|---|
protected FeatureConverter |
TensorFlowModel.featureConverter |
| Modifier and Type | Method and Description |
|---|---|
static <T extends Output<T>> |
TensorFlowSavedModelExternalModel.createTensorflowModel(OutputFactory<T> factory,
Map<String,Integer> featureMapping,
Map<T,Integer> outputMapping,
String outputName,
FeatureConverter featureConverter,
OutputConverter<T> outputConverter,
String bundleDirectory)
Creates a TensorflowSavedModelExternalModel by loading in a
SavedModelBundle. |
static <T extends Output<T>> |
TensorFlowFrozenExternalModel.createTensorflowModel(OutputFactory<T> factory,
Map<String,Integer> featureMapping,
Map<T,Integer> outputMapping,
String inputName,
String outputName,
FeatureConverter featureConverter,
OutputConverter<T> outputConverter,
String filename)
Creates a TensorflowFrozenExternalModel by loading in a frozen graph.
|
| Constructor and Description |
|---|
TensorFlowModel(String name,
ModelProvenance provenance,
ImmutableFeatureMap featureIDMap,
ImmutableOutputInfo<T> outputIDInfo,
org.tensorflow.proto.framework.GraphDef trainedGraphDef,
int batchSize,
String outputName,
FeatureConverter featureConverter,
OutputConverter<T> outputConverter)
Builds a TFModel.
|
TensorFlowTrainer(org.tensorflow.proto.framework.GraphDef graphDef,
String outputName,
GradientOptimiser optimiser,
Map<String,Float> gradientParams,
FeatureConverter featureConverter,
OutputConverter<T> outputConverter,
int trainBatchSize,
int epochs,
int testBatchSize,
int loggingInterval)
Constructs a Trainer for a TensorFlow graph.
|
TensorFlowTrainer(org.tensorflow.proto.framework.GraphDef graphDef,
String outputName,
GradientOptimiser optimiser,
Map<String,Float> gradientParams,
FeatureConverter featureConverter,
OutputConverter<T> outputConverter,
int trainBatchSize,
int epochs,
int testBatchSize,
int loggingInterval,
Path checkpointPath)
Constructs a Trainer for a TensorFlow graph.
|
TensorFlowTrainer(org.tensorflow.Graph graph,
String outputName,
GradientOptimiser optimiser,
Map<String,Float> gradientParams,
FeatureConverter featureConverter,
OutputConverter<T> outputConverter,
int trainBatchSize,
int epochs,
int testBatchSize,
int loggingInterval)
Constructs a Trainer for a TensorFlow graph.
|
TensorFlowTrainer(org.tensorflow.Graph graph,
String outputName,
GradientOptimiser optimiser,
Map<String,Float> gradientParams,
FeatureConverter featureConverter,
OutputConverter<T> outputConverter,
int trainBatchSize,
int epochs,
int testBatchSize,
int loggingInterval,
Path checkpointPath)
Constructs a Trainer for a TensorFlow graph.
|
TensorFlowTrainer(Path graphPath,
String outputName,
GradientOptimiser optimiser,
Map<String,Float> gradientParams,
FeatureConverter featureConverter,
OutputConverter<T> outputConverter,
int trainBatchSize,
int epochs,
int testBatchSize,
int loggingInterval)
Constructs a Trainer for a TensorFlow graph.
|
TensorFlowTrainer(Path graphPath,
String outputName,
GradientOptimiser optimiser,
Map<String,Float> gradientParams,
FeatureConverter featureConverter,
OutputConverter<T> outputConverter,
int trainBatchSize,
int epochs,
int testBatchSize,
int loggingInterval,
Path checkpointPath)
Constructs a Trainer for a TensorFlow graph.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.