Package 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
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description UIListenerbuild()UIListener.BuilderfileMode(UIListener.FileMode fileMode)UIListener.Builderhistograms(int frequency, UIListener.HistogramType... types)UIListener.BuilderlearningRate(int frequency)UIListener.BuilderperformanceStats(int frequency)UIListener.BuilderplotLosses(int frequency)UIListener.BuilderprofileOps(int frequency)UIListener.BuildertestEvaluation(UIListener.TestEvaluation testEvalConfig)UIListener.BuildertrainAccuracy(String name, int labelIdx)UIListener.BuildertrainEvalFrequency(int trainEvalFrequency)UIListener.BuildertrainEvaluationMetrics(String name, int labelIdx, Evaluation.Metric... metrics)UIListener.BuildertrainF1(String name, int labelIdx)UIListener.BuilderupdateRatios(int frequency)UIListener.BuilderupdateRatios(int frequency, UIListener.UpdateRatio ratioType)
-
-
-
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()
-
-