Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

...

  • New time series models

  • New time series loss functions

  • Bug fixes

  • Meta data embedding models.

  • Documentation updates and fixes

  • Increasing test coverage

Code Style

We use Flake8 and mypy. Currently, we are a PyTorch only repository in terms of model frameworks.

...

All PRs require at least one approving review. Generally this review should be either from Isaac Godfried kriti mahajan or another core contributor. Please feel free to request to a review.

Getting setup locally

To get setup local you can install our repository in development mode from the master branch (or whatever branch you are using). To do this run

Code Block
cd flow-forecast
python setup.py develop

Adding models

Models generally require four things:

...

  • Models that take parameters other than X and metadata in forward_params.

  • Models that return a tensor not in the shape (batch_size, seq_len, n_results)

Please plan with the repository owners before unde Isaac Godfried before undertaking these more complicated models.

Adding Loss Functions