kdafnis/controllabledl

Controllable deep learning (controllabledl)

This is not an officially supported Google product.

Dependencies

• python 3.7
• pytorch 1.6.0

Double pendulum system

Data

For the double pendulum dataset, please check pendulum-system/train.py to see how data is generated.

Command

$ python train.py --model_type dnn-crr-autoscale

M5 forecasting task

• Task description and dataset: https://www.kaggle.com/c/m5-forecasting-accuracy/overview

Healthcare task

• Task description and dataset: https://www.kaggle.com/sulianova/cardiovascular-disease-dataset