org.nd4j.autodiff.listeners.impl

Class UIListener.Builder

java.lang.Object

org.nd4j.autodiff.listeners.impl.UIListener.Builder

Enclosing class:

UIListener

public static class UIListener.Builder
extends Object

Constructor Summary

Constructors

Constructor and Description
Builder(@NonNull File logFile)

Method Summary

Modifier and Type	Method and Description
UIListener	build()
UIListener.Builder	fileMode(UIListener.FileMode fileMode)
UIListener.Builder	histograms(int frequency, UIListener.HistogramType... types)
UIListener.Builder	learningRate(int frequency)
UIListener.Builder	performanceStats(int frequency)
UIListener.Builder	plotLosses(int frequency)
UIListener.Builder	profileOps(int frequency)
UIListener.Builder	testEvaluation(UIListener.TestEvaluation testEvalConfig)
UIListener.Builder	trainAccuracy(String name, int labelIdx)
UIListener.Builder	trainEvalFrequency(int trainEvalFrequency)
UIListener.Builder	trainEvaluationMetrics(String name, int labelIdx, Evaluation.Metric... metrics)
UIListener.Builder	trainF1(String name, int labelIdx)
UIListener.Builder	updateRatios(int frequency)
UIListener.Builder	updateRatios(int frequency, UIListener.UpdateRatio ratioType)

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder(@NonNull
               @NonNull File logFile)

Method Detail

fileMode

public UIListener.Builder fileMode(UIListener.FileMode fileMode)

plotLosses

public UIListener.Builder plotLosses(int frequency)

performanceStats

public UIListener.Builder performanceStats(int frequency)

trainEvaluationMetrics

public UIListener.Builder trainEvaluationMetrics(String name,
                                                 int labelIdx,
                                                 Evaluation.Metric... metrics)

trainAccuracy

public UIListener.Builder trainAccuracy(String name,
                                        int labelIdx)

trainF1

public UIListener.Builder trainF1(String name,
                                  int labelIdx)

trainEvalFrequency

public UIListener.Builder trainEvalFrequency(int trainEvalFrequency)

updateRatios

public UIListener.Builder updateRatios(int frequency)

updateRatios

public UIListener.Builder updateRatios(int frequency,
                                       UIListener.UpdateRatio ratioType)

histograms

public UIListener.Builder histograms(int frequency,
                                     UIListener.HistogramType... types)

profileOps

public UIListener.Builder profileOps(int frequency)

testEvaluation

public UIListener.Builder testEvaluation(UIListener.TestEvaluation testEvalConfig)

learningRate

public UIListener.Builder learningRate(int frequency)

build

public UIListener build()