QRNN_RNN_net | R Documentation |
A recurrent neural network as a torch::nn_module
,
designed for quantile regression.
QRNN_RNN_net(
type = c("lstm", "gru"),
nb_input_features,
hidden_size,
num_layers = 1,
dropout = 0
)
type |
the type of recurrent architecture, can be one of |
nb_input_features |
the input size (i.e. the number of features), |
the dimension of the hidden latent state variables in the recurrent network, | |
num_layers |
the number of recurrent layers, |
dropout |
probability parameter for dropout before each hidden layer for regularization during training. |
The constructor allows specifying:
the type of recurrent architecture, can be one of "lstm"
(default) or "gru"
,
the input size (i.e. the number of features),
the dimension of the hidden latent state variables in the recurrent network,
the number of recurrent layers,
probability parameter for dropout before each hidden layer for regularization during training.
The specified recurrent QRN as a torch::nn_module
.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.