Skip to content

Instantly share code, notes, and snippets.

@hrzn
Created August 9, 2021 16:46
Show Gist options
  • Save hrzn/90f37efe221248e487f50d90e404944a to your computer and use it in GitHub Desktop.
Save hrzn/90f37efe221248e487f50d90e404944a to your computer and use it in GitHub Desktop.
from darts.models import RNNModel
rnn_rain = RNNModel(input_chunk_length=30,
training_length=40,
n_rnn_layers=2)
rnn_rain.fit(flow_train,
future_covariates=rainfalls,
epochs=100,
verbose=True)
eval_model(rnn_rain,
future_covariates=rainfalls)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment