hookeai.model_architectures.rnn_base_model.model.gru_model

Multi-layer gated recurrent unit (GRU) recurrent neural network model.

Classes

GRURNNModel(torch.nn.Module)

Multi-layer gated recurrent unit (GRU) recurrent neural network model.

Functions

init_data_scalers(model)	Initialize model data scalers.
save_model_state(model[, state_type, epoch, ...])	Save model state to file.

Classes

GRU(input_size, hidden_size[, num_layers, ...])	Multi-layer gated recurrent unit (GRU) recurrent neural network model.
GRURNNModel(n_features_in, n_features_out, ...)	Multi-layer gated recurrent unit (GRU) recurrent neural network model.