Package org.nd4j.autodiff.listeners

Class ListenerEvaluations.Builder

java.lang.Object

org.nd4j.autodiff.listeners.ListenerEvaluations.Builder

Enclosing class:

ListenerEvaluations

public static class ListenerEvaluations.Builder
extends Object

Constructor Summary

Constructors

Constructor	Description
Builder()

Method Summary

Modifier and Type	Method	Description
ListenerEvaluations.Builder	addEvaluations(boolean validation, @NonNull String variableName, int labelIndex, @NonNull IEvaluation... evaluations)	Add requested evaluations for a parm/variable, for either training or validation
ListenerEvaluations	build()
ListenerEvaluations.Builder	trainEvaluation(@NonNull String variableName, int labelIndex, @NonNull IEvaluation... evaluations)	Add requested training evaluations for a parm/variable
ListenerEvaluations.Builder	trainEvaluation(@NonNull SDVariable variable, int labelIndex, @NonNull IEvaluation... evaluations)	Add requested training evaluations for a parm/variable
ListenerEvaluations.Builder	validationEvaluation(@NonNull String variableName, int labelIndex, @NonNull IEvaluation... evaluations)	Add requested validation evaluations for a parm/variable
ListenerEvaluations.Builder	validationEvaluation(@NonNull SDVariable variable, int labelIndex, @NonNull IEvaluation... evaluations)	Add requested validation evaluations for a parm/variable

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

trainEvaluation

public ListenerEvaluations.Builder trainEvaluation(@NonNull
                                                   @NonNull String variableName,
                                                   int labelIndex,
                                                   @NonNull
                                                   @NonNull IEvaluation... evaluations)

Add requested training evaluations for a parm/variable

Parameters:

variableName - The variable to evaluate

labelIndex - The index of the label to evaluate against

evaluations - The evaluations to run

trainEvaluation

public ListenerEvaluations.Builder trainEvaluation(@NonNull
                                                   @NonNull SDVariable variable,
                                                   int labelIndex,
                                                   @NonNull
                                                   @NonNull IEvaluation... evaluations)

Add requested training evaluations for a parm/variable

Parameters:

variable - The variable to evaluate

labelIndex - The index of the label to evaluate against

evaluations - The evaluations to run

validationEvaluation

public ListenerEvaluations.Builder validationEvaluation(@NonNull
                                                        @NonNull String variableName,
                                                        int labelIndex,
                                                        @NonNull
                                                        @NonNull IEvaluation... evaluations)

Add requested validation evaluations for a parm/variable

Parameters:

variableName - The variable to evaluate

labelIndex - The index of the label to evaluate against

evaluations - The evaluations to run

validationEvaluation

public ListenerEvaluations.Builder validationEvaluation(@NonNull
                                                        @NonNull SDVariable variable,
                                                        int labelIndex,
                                                        @NonNull
                                                        @NonNull IEvaluation... evaluations)

Add requested validation evaluations for a parm/variable

Parameters:

variable - The variable to evaluate

labelIndex - The index of the label to evaluate against

evaluations - The evaluations to run

addEvaluations

public ListenerEvaluations.Builder addEvaluations(boolean validation,
                                                  @NonNull
                                                  @NonNull String variableName,
                                                  int labelIndex,
                                                  @NonNull
                                                  @NonNull IEvaluation... evaluations)

Add requested evaluations for a parm/variable, for either training or validation

Parameters:

validation - Whether to add these evaluations as validation or training

variableName - The variable to evaluate

labelIndex - The index of the label to evaluate against

evaluations - The evaluations to run

build

public ListenerEvaluations build()