search

ServiceNow / TACTiS

TACTiS-2: Better, Faster, Simpler Attentional Copulas for Multivariate Time Series, from ServiceNow Research
Apache License 2.0
110 stars 19 forks source link

TACTiS-2 code - first version #22

Closed ashok-arjun closed 8 months ago

ashok-arjun commented 8 months ago

This is the first version of the code of TACTiS-2. This version contains the TACTiS-2 model and allows for training the model for any dataset using the train.py file.

This version does not support:

  1. An interface for evaluating the model and getting metrics
  2. Obtaining the validation negative log-likelihood without bagging
  3. Demo Jupyter notebooks
  4. Training options such as batch size search

These functionalities will be added soon in future versions.

ashok-arjun commented 8 months ago

Additionally modified train.py to rename checkpoint_path to checkpoint_dir for better naming.