Package org.apache.wayang.basic.operators

Class DLTrainingOperator.Option

java.lang.Object

org.apache.wayang.basic.operators.DLTrainingOperator.Option

Enclosing class:

DLTrainingOperator<X,Y>

public static class DLTrainingOperator.Option
extends Object

Constructor Summary

Constructors

Constructor	Description
Option(Op criterion, Optimizer optimizer, int batchSize, int epoch)

Method Summary

Modifier and Type	Method	Description
Op	getAccuracyCalculation()
int	getBatchSize()
Op	getCriterion()
int	getEpoch()
Optimizer	getOptimizer()
void	setAccuracyCalculation(Op accuracyCalculation)

Methods inherited from class java.lang.Object

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

Constructor Details

Option

public Option(Op criterion,
 Optimizer optimizer,
 int batchSize,
 int epoch)

Method Details

getCriterion

public Op getCriterion()

getOptimizer

public Optimizer getOptimizer()

getBatchSize

public int getBatchSize()

getEpoch

public int getEpoch()

setAccuracyCalculation

public void setAccuracyCalculation(Op accuracyCalculation)

getAccuracyCalculation

public Op getAccuracyCalculation()