Package io.github.kirstenali.deepj.training

Class SupervisedTraining

java.lang.Object

io.github.kirstenali.deepj.training.SupervisedTraining

public final class SupervisedTraining
extends Object

Helpers to train classic Tensor->Tensor supervised models (e.g., FNN) using the unified Trainer wrapper.

Method Summary

Modifier and Type	Method	Description
static Trainer	trainer(Layer model, LossFunction lossFn, ParameterOptimizer opt, Tensor xAll, Tensor yAll, long seed)

Methods inherited from class java.lang.Object

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

Method Details

trainer

public static Trainer trainer(Layer model,
 LossFunction lossFn,
 ParameterOptimizer opt,
 Tensor xAll,
 Tensor yAll,
 long seed)