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 |
TensorFlowCheckpointModel<T extends Output<T>>
This model encapsulates a simple model with an input feed dict,
and produces a single output tensor.
|
class |
TensorFlowNativeModel<T extends Output<T>>
This model encapsulates a TensorFlow model running in graph mode with a single tensor output.
|
Modifier and Type | Method and Description |
---|---|
TensorFlowModel<T> |
TensorFlowTrainer.train(Dataset<T> examples) |
TensorFlowModel<T> |
TensorFlowTrainer.train(Dataset<T> examples,
Map<String,com.oracle.labs.mlrg.olcut.provenance.Provenance> runProvenance) |
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.