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

Modifier and Type	Method	Description
ListenerVariables	build()
ListenerVariables.Builder	evaluationVariables(@NonNull String... variables)	Add required variables for evaluation
ListenerVariables.Builder	evaluationVariables(@NonNull SDVariable... variables)	Add required variables for evaluation
ListenerVariables.Builder	inferenceVariables(@NonNull String... variables)	Add required variables for inference
ListenerVariables.Builder	inferenceVariables(@NonNull SDVariable... variables)	Add required variables for inference
ListenerVariables.Builder	requireVariables(@NonNull Operation op, @NonNull String... variables)	Add required variables for the specified op
ListenerVariables.Builder	requireVariables(@NonNull Operation op, @NonNull SDVariable... variables)	Add required variables for the specified op
ListenerVariables.Builder	trainingVariables(@NonNull String... variables)	Add required variables for training
ListenerVariables.Builder	trainingVariables(@NonNull SDVariable... variables)	Add required variables for training
ListenerVariables.Builder	validationVariables(@NonNull String... variables)	Add required variables for validation
ListenerVariables.Builder	validationVariables(@NonNull SDVariable... variables)	Add required variables for validation

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

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()