Package io.github.kirstenali.deepj.nn

Class NdSupervisedTraining

java.lang.Object

io.github.kirstenali.deepj.nn.NdSupervisedTraining

public final class NdSupervisedTraining
extends Object

First-axis minibatching for images, sequences and matrices, using DeepJ's existing Trainer.

Method Summary

Modifier and Type	Method	Description
static Trainer	trainer(Layer model, LossFunction loss, ParameterOptimizer optimizer, Tensor inputs, Tensor targets, 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 loss,
 ParameterOptimizer optimizer,
 Tensor inputs,
 Tensor targets,
 long seed)