Skip to content

Instantly share code, notes, and snippets.

@hrzn

hrzn/river_flow_brnn_no_cov.py

Created August 9, 2021 15:46
Show Gist options
  • Save hrzn/2e9614e1946647dd0bb704f0b14b9a81 to your computer and use it in GitHub Desktop.
Save hrzn/2e9614e1946647dd0bb704f0b14b9a81 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
river_flow_brnn_no_cov.py
from darts.models import BlockRNNModel
brnn_no_cov = BlockRNNModel(input_chunk_length=30,
output_chunk_length=10,
n_rnn_layers=2)
brnn_no_cov.fit(flow_train,
epochs=100,
verbose=True)
eval_model(brnn_no_cov)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment