Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[DOC] docstring for SimpleRNNClassifier #4572

Merged
merged 5 commits into from May 23, 2023
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
30 changes: 30 additions & 0 deletions sktime/classification/deep_learning/rnn.py
Expand Up @@ -19,6 +19,36 @@

class SimpleRNNClassifier(BaseDeepClassifier):
"""Simple recurrent neural network.

Parameters
----------
should inherited fields be listed here?
wasup-yash marked this conversation as resolved.
Show resolved Hide resolved
n_epochs : int, default = 2000
the number of epochs to train the model
batch_size : int, default = 16
the number of samples per gradient update.
kernel_size : int, default = 7
the length of the 1D convolution window
avg_pool_size : int, default = 3
size of the average pooling windows
n_conv_layers : int, default = 2
the number of convolutional plus average pooling layers
filter_sizes : array of shape (n_conv_layers) default = [6, 12]
random_state : int or None, default=None
Seed for random number generation.
verbose : boolean, default = False
whether to output extra information
loss : string, default="mean_squared_error"
fit parameter for the keras model
optimizer : keras.optimizer, default=keras.optimizers.Adam(),
metrics : list of strings, default=["accuracy"],
activation : string or a tf callable, default="sigmoid"
Activation function used in the output linear layer.
List of available activation functions: https://keras.io/api/layers/activations/
use_bias : boolean, default = True
whether the layer uses a bias vector.
optimizer : keras.optimizers object, default = Adam(lr=0.01)
specify the optimizer and the learning rate to be used.

References
----------
Expand Down
30 changes: 30 additions & 0 deletions sktime/regression/deep_learning/rnn.py
Expand Up @@ -20,6 +20,36 @@
class SimpleRNNRegressor(BaseDeepRegressor):
"""
Simple recurrent neural network.

Parameters
----------
should inherited fields be listed here?
n_epochs : int, default = 2000
the number of epochs to train the model
batch_size : int, default = 16
the number of samples per gradient update.
kernel_size : int, default = 7
the length of the 1D convolution window
avg_pool_size : int, default = 3
size of the average pooling windows
n_conv_layers : int, default = 2
the number of convolutional plus average pooling layers
filter_sizes : array of shape (n_conv_layers) default = [6, 12]
random_state : int or None, default=None
Seed for random number generation.
verbose : boolean, default = False
whether to output extra information
loss : string, default="mean_squared_error"
fit parameter for the keras model
optimizer : keras.optimizer, default=keras.optimizers.Adam(),
metrics : list of strings, default=["accuracy"],
activation : string or a tf callable, default="sigmoid"
Activation function used in the output linear layer.
List of available activation functions:https://keras.io/api/layers/activations/
use_bias : boolean, default = True
whether the layer uses a bias vector.
optimizer : keras.optimizers object, default = Adam(lr=0.01)
specify the optimizer and the learning rate to be used.

References
----------
Expand Down