Class EvaluationConfig
- java.lang.Object
-
- org.nd4j.autodiff.samediff.config.EvaluationConfig
-
public class EvaluationConfig extends Object
-
-
Constructor Summary
Constructors Constructor Description EvaluationConfig(@NonNull SameDiff sd)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description EvaluationConfigdata(@NonNull DataSetIterator data)Set the data to evaluate on.EvaluationConfigdata(@NonNull MultiDataSetIterator data)Set the data to evaluate on.EvaluationConfigevaluate(@NonNull String param, int labelIndex, @NonNull IEvaluation... evaluations)Add evaluations to be preformed on a specified variable, and set that variable's label index.EvaluationConfigevaluate(@NonNull String param, @NonNull IEvaluation... evaluations)Add evaluations to be preformed on a specified variable, without setting a label index.EvaluationConfigevaluate(@NonNull SDVariable variable, int labelIndex, @NonNull IEvaluation... evaluations)EvaluationConfigevaluate(@NonNull SDVariable variable, @NonNull IEvaluation... evaluations)EvaluationRecordexec()Run the evaluation.EvaluationConfiglabelIndex(@NonNull String param, int labelIndex)Set the label index for a parameterEvaluationConfiglabelIndex(@NonNull SDVariable variable, int labelIndex)EvaluationConfiglisteners(@NonNull Listener... listeners)Add listeners for this operation
-
-
-
Constructor Detail
-
EvaluationConfig
public EvaluationConfig(@NonNull @NonNull SameDiff sd)
-
-
Method Detail
-
evaluate
public EvaluationConfig evaluate(@NonNull @NonNull String param, int labelIndex, @NonNull @NonNull IEvaluation... evaluations)
Add evaluations to be preformed on a specified variable, and set that variable's label index. Setting a label index is required if using a MultiDataSetIterator.- Parameters:
param- The param to evaluatelabelIndex- The label index of that parameterevaluations- The evaluations to preform
-
evaluate
public EvaluationConfig evaluate(@NonNull @NonNull SDVariable variable, int labelIndex, @NonNull @NonNull IEvaluation... evaluations)
-
evaluate
public EvaluationConfig evaluate(@NonNull @NonNull String param, @NonNull @NonNull IEvaluation... evaluations)
Add evaluations to be preformed on a specified variable, without setting a label index. Setting a label index (which is not done here) is required if using a MultiDataSetIterator.- Parameters:
param- The param to evaluateevaluations- The evaluations to preform
-
evaluate
public EvaluationConfig evaluate(@NonNull @NonNull SDVariable variable, @NonNull @NonNull IEvaluation... evaluations)
-
labelIndex
public EvaluationConfig labelIndex(@NonNull @NonNull String param, int labelIndex)
Set the label index for a parameter
-
labelIndex
public EvaluationConfig labelIndex(@NonNull @NonNull SDVariable variable, int labelIndex)
-
listeners
public EvaluationConfig listeners(@NonNull @NonNull Listener... listeners)
Add listeners for this operation
-
data
public EvaluationConfig data(@NonNull @NonNull MultiDataSetIterator data)
Set the data to evaluate on. Setting a label index for each variable to evaluate is required
-
data
public EvaluationConfig data(@NonNull @NonNull DataSetIterator data)
Set the data to evaluate on. Setting a label index for each variable to evaluate is NOT required (since there is only one input)
-
exec
public EvaluationRecord exec()
Run the evaluation. Note that the evaluations in the returnedEvaluationRecordare the evaluations set usingevaluate(String, int, IEvaluation[]), it does not matter which you use to access results.- Returns:
- The specified listeners, in an
EvaluationRecordfor easy access.
-
-