Package org.nd4j.autodiff.listeners
Class ListenerVariables.Builder
- java.lang.Object
-
- org.nd4j.autodiff.listeners.ListenerVariables.Builder
-
- Enclosing class:
- ListenerVariables
public static class ListenerVariables.Builder extends Object
-
-
Constructor Summary
Constructors Constructor Description Builder()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ListenerVariablesbuild()ListenerVariables.BuilderevaluationVariables(@NonNull String... variables)Add required variables for evaluationListenerVariables.BuilderevaluationVariables(@NonNull SDVariable... variables)Add required variables for evaluationListenerVariables.BuilderinferenceVariables(@NonNull String... variables)Add required variables for inferenceListenerVariables.BuilderinferenceVariables(@NonNull SDVariable... variables)Add required variables for inferenceListenerVariables.BuilderrequireVariables(@NonNull Operation op, @NonNull String... variables)Add required variables for the specified opListenerVariables.BuilderrequireVariables(@NonNull Operation op, @NonNull SDVariable... variables)Add required variables for the specified opListenerVariables.BuildertrainingVariables(@NonNull String... variables)Add required variables for trainingListenerVariables.BuildertrainingVariables(@NonNull SDVariable... variables)Add required variables for trainingListenerVariables.BuildervalidationVariables(@NonNull String... variables)Add required variables for validationListenerVariables.BuildervalidationVariables(@NonNull SDVariable... variables)Add required variables for validation
-
-
-
Method Detail
-
requireVariables
public ListenerVariables.Builder requireVariables(@NonNull @NonNull Operation op, @NonNull @NonNull String... variables)
Add required variables for the specified op- Parameters:
op- The op to require the variable for
-
requireVariables
public ListenerVariables.Builder requireVariables(@NonNull @NonNull Operation op, @NonNull @NonNull SDVariable... variables)
Add required variables for the specified op- Parameters:
op- The op to require the variable for
-
trainingVariables
public ListenerVariables.Builder trainingVariables(@NonNull @NonNull String... variables)
Add required variables for training
-
trainingVariables
public ListenerVariables.Builder trainingVariables(@NonNull @NonNull SDVariable... variables)
Add required variables for training
-
validationVariables
public ListenerVariables.Builder validationVariables(@NonNull @NonNull String... variables)
Add required variables for validation
-
validationVariables
public ListenerVariables.Builder validationVariables(@NonNull @NonNull SDVariable... variables)
Add required variables for validation
-
inferenceVariables
public ListenerVariables.Builder inferenceVariables(@NonNull @NonNull String... variables)
Add required variables for inference
-
inferenceVariables
public ListenerVariables.Builder inferenceVariables(@NonNull @NonNull SDVariable... variables)
Add required variables for inference
-
evaluationVariables
public ListenerVariables.Builder evaluationVariables(@NonNull @NonNull String... variables)
Add required variables for evaluation
-
evaluationVariables
public ListenerVariables.Builder evaluationVariables(@NonNull @NonNull SDVariable... variables)
Add required variables for evaluation
-
build
public ListenerVariables build()
-
-