Package io.github.kirstenali.deepj.training

Class Trainer

java.lang.Object

io.github.kirstenali.deepj.training.Trainer

public final class Trainer
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	Trainer.StepFunction
static interface	Trainer.StepHook

Constructor Summary

Constructors

Constructor	Description
Trainer(Trainer.StepFunction stepFn)

Method Summary

Modifier and Type	Method	Description
TrainingResult	train(int maxSteps, int batchSize, int logEvery, float emaBeta, Float targetEmaLoss)
TrainingResult	train(int maxSteps, int batchSize, int logEvery, float emaBeta, Float targetEmaLoss, int releaseEverySteps)
TrainingResult	train(int maxSteps, int batchSize, int logEvery, float emaBeta, Float targetEmaLoss, int releaseEverySteps, Trainer.StepHook stepHook)
TrainingResult	train(int maxSteps, int batchSize, int logEvery, float emaBeta, Float targetEmaLoss, int releaseEverySteps, Trainer.StepHook stepHook, TrainingProgress progress)
TrainingResult	train(int maxSteps, int batchSize, int logEvery, float emaBeta, Float targetEmaLoss, Trainer.StepHook stepHook)
float	trainStep(int batchSize)

Methods inherited from class java.lang.Object

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

Constructor Details

Trainer

public Trainer(Trainer.StepFunction stepFn)

Method Details

trainStep

public float trainStep(int batchSize)

train

public TrainingResult train(int maxSteps,
 int batchSize,
 int logEvery,
 float emaBeta,
 Float targetEmaLoss)

train

public TrainingResult train(int maxSteps,
 int batchSize,
 int logEvery,
 float emaBeta,
 Float targetEmaLoss,
 int releaseEverySteps)

train

public TrainingResult train(int maxSteps,
 int batchSize,
 int logEvery,
 float emaBeta,
 Float targetEmaLoss,
 Trainer.StepHook stepHook)

train

public TrainingResult train(int maxSteps,
 int batchSize,
 int logEvery,
 float emaBeta,
 Float targetEmaLoss,
 int releaseEverySteps,
 Trainer.StepHook stepHook)

train

public TrainingResult train(int maxSteps,
 int batchSize,
 int logEvery,
 float emaBeta,
 Float targetEmaLoss,
 int releaseEverySteps,
 Trainer.StepHook stepHook,
 TrainingProgress progress)