Tensorflow
Tensorflow interop Key features and brief samples.
Nd4j allows execution of models via javacpp's tensorflow bindings using nd4j's INDArray as a data structure. Leveraging the nd4j-tensorflow interop is fairly simple.
Note, this is based on tensorflow 1.x. TF java 2 will be coming at a later time.
A tensorflow model can be loaded and executed as follows:
This outputs a result with a map of output names to the ndarray result from tensorflow.
In maven, add the following dependency:
Last updated
Was this helpful?