Package org.apache.wayang.tensorflow.model

Class TensorflowModel

java.lang.Object

org.apache.wayang.basic.model.DLModel

org.apache.wayang.tensorflow.model.TensorflowModel

All Implemented Interfaces:

Serializable, AutoCloseable, Model

public class TensorflowModel
extends DLModel
implements AutoCloseable

See Also:

• Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.wayang.basic.model.DLModel

DLModel.Builder

Field Summary

Fields inherited from class org.apache.wayang.basic.model.DLModel

out

Constructor Summary

Constructors

Constructor	Description
TensorflowModel(DLModel model, Op criterion, Optimizer optimizer, Op accuracyCalculation)

Method Summary

Modifier and Type	Method	Description
void	close()
Op	getAccuracyCalculation()
Op	getCriterion()
Optimizer	getOptimizer()
<XT extends org.tensorflow.ndarray.NdArray<?>, PT extends org.tensorflow.ndarray.NdArray<?> & org.tensorflow.types.family.TType> PT	predict(XT x)
<XT extends org.tensorflow.ndarray.NdArray<?>, YT extends org.tensorflow.ndarray.NdArray<?>> void	train(XT x, YT y, int epoch, int batchSize)

Methods inherited from class org.apache.wayang.basic.model.DLModel

getOut

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

TensorflowModel

public TensorflowModel(DLModel model,
 Op criterion,
 Optimizer optimizer,
 Op accuracyCalculation)

Method Details

train

public <XT extends org.tensorflow.ndarray.NdArray<?>,
YT extends org.tensorflow.ndarray.NdArray<?>>
void train(XT x,
 YT y,
 int epoch,
 int batchSize)

predict

public <XT extends org.tensorflow.ndarray.NdArray<?>,
PT extends org.tensorflow.ndarray.NdArray<?> & org.tensorflow.types.family.TType>
PT predict(XT x)

getCriterion

public Op getCriterion()

getOptimizer

public Optimizer getOptimizer()

getAccuracyCalculation

public Op getAccuracyCalculation()

close

public void close()

Specified by:

close in interface AutoCloseable