fynance.models.neural_network.BaseNeuralNet.train_on

BaseNeuralNet.train_on(X, y)

Trains the neural network model.

Parameters:

X, y : torch.Tensor

Respectively inputs and outputs to train model.

Returns:

torch.nn.modules.loss

Loss outputs.