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 <T extends Output<T>> |
TensorflowExternalModel.createTensorflowModel(OutputFactory<T> factory,
Map<String,Integer> featureMapping,
Map<T,Integer> outputMapping,
String inputName,
String outputName,
ExampleTransformer<T> featureTransformer,
OutputTransformer<T> outputTransformer,
String filename)
Creates a TensorflowExternalModel by loading in a frozen graph.
|
Constructor and Description |
---|
TensorflowCheckpointTrainer(Path graphPath,
Path checkpointRootPath,
ExampleTransformer<T> exampleTransformer,
OutputTransformer<T> outputTransformer,
int minibatchSize,
int epochs)
Builds a trainer using the supplied graph and arguments.
|
TensorflowTrainer(byte[] graphDef,
ExampleTransformer<T> exampleTransformer,
OutputTransformer<T> outputTransformer,
int minibatchSize,
int epochs,
int testBatchSize)
Constructs a Trainer for a tensorflow graph.
|
TensorflowTrainer(Path graphPath,
ExampleTransformer<T> exampleTransformer,
OutputTransformer<T> outputTransformer,
int minibatchSize,
int epochs,
int testBatchSize)
Constructs a Trainer for a tensorflow graph.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.