fynance.models.recurrent_neural_network.GatedRecurrentUnit.train_on

GatedRecurrentUnit.train_on(X, y, H)

Trains the neural network model.

Parameters:

X, y, H : torch.Tensor

Respectively inputs, outputs and states to train model.

Returns:

torch.nn.modules.loss

Loss outputs.

torch.Tensor

Updated states of the model.