├── .flake8 ├── .github └── workflows │ └── pre-commit.yml ├── .gitignore ├── .isort.cfg ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── assets ├── Log-NCDE.png └── time_vs_acc.png ├── data_dir ├── dataloaders.py ├── datasets.py ├── download_uea.py ├── generate_coeffs.py ├── generate_paths.py ├── hall_set.py ├── process_ppg.py ├── process_uea.py └── toy_dataset.py ├── experiment_configs ├── repeats │ ├── S5 │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ ├── SelfRegulationSCP2.json │ │ └── ppg.json │ ├── S6 │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ ├── SelfRegulationSCP2.json │ │ └── ppg.json │ ├── bd_linear_ncde │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ └── SelfRegulationSCP2.json │ ├── dense_linear_ncde │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ └── SelfRegulationSCP2.json │ ├── diagonal_dense_linear_ncde │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ └── SelfRegulationSCP2.json │ ├── diagonal_linear_ncde │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ └── SelfRegulationSCP2.json │ ├── dplr_linear_ncde │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ └── SelfRegulationSCP2.json │ ├── log_ncde │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ ├── SelfRegulationSCP2.json │ │ └── ppg.json │ ├── lru │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ ├── SelfRegulationSCP2.json │ │ └── ppg.json │ ├── mamba │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ ├── SelfRegulationSCP2.json │ │ └── ppg.json │ ├── ncde │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ ├── SelfRegulationSCP2.json │ │ └── ppg.json │ ├── nrde │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ ├── SelfRegulationSCP2.json │ │ └── ppg.json │ ├── sparse_linear_ncde │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ └── SelfRegulationSCP2.json │ └── wh_linear_ncde │ │ ├── EigenWorms.json │ │ ├── EthanolConcentration.json │ │ ├── Heartbeat.json │ │ ├── MotorImagery.json │ │ ├── SelfRegulationSCP1.json │ │ └── SelfRegulationSCP2.json └── toy │ ├── S5 │ ├── signature1.json │ ├── signature2.json │ ├── signature3.json │ └── signature4.json │ ├── S6 │ ├── signature1.json │ ├── signature2.json │ ├── signature3.json │ └── signature4.json │ ├── log_ncde │ ├── signature1.json │ ├── signature2.json │ ├── signature3.json │ └── signature4.json │ ├── lru │ ├── signature1.json │ ├── signature2.json │ ├── signature3.json │ └── signature4.json │ ├── mamba │ ├── signature1.json │ ├── signature2.json │ ├── signature3.json │ └── signature4.json │ ├── ncde │ ├── signature1.json │ ├── signature2.json │ ├── signature3.json │ └── signature4.json │ └── nrde │ ├── signature1.json │ ├── signature2.json │ ├── signature3.json │ └── signature4.json ├── models ├── LRU.py ├── LinearNeuralCDEs.py ├── LogNeuralCDEs.py ├── NeuralCDEs.py ├── RNN.py ├── S5.py └── generate_model.py ├── results ├── analyse_results.py ├── hyperparam_info_for_table.json ├── logsig_length.py ├── memory_time.py ├── memory_time_results.json ├── paper_outputs.zip ├── plot_toy.py ├── produce_hypparam_table.py └── time_vs_acc.py ├── run_experiment.py ├── simple_example.ipynb ├── torch_experiments ├── jax_dataset.py ├── s6_recurrence.py └── train.py └── train.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/.flake8 -------------------------------------------------------------------------------- /.github/workflows/pre-commit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/.github/workflows/pre-commit.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/.gitignore -------------------------------------------------------------------------------- /.isort.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/.isort.cfg -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/README.md -------------------------------------------------------------------------------- /assets/Log-NCDE.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/assets/Log-NCDE.png -------------------------------------------------------------------------------- /assets/time_vs_acc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/assets/time_vs_acc.png -------------------------------------------------------------------------------- /data_dir/dataloaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/data_dir/dataloaders.py -------------------------------------------------------------------------------- /data_dir/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/data_dir/datasets.py -------------------------------------------------------------------------------- /data_dir/download_uea.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/data_dir/download_uea.py -------------------------------------------------------------------------------- /data_dir/generate_coeffs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/data_dir/generate_coeffs.py -------------------------------------------------------------------------------- /data_dir/generate_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/data_dir/generate_paths.py -------------------------------------------------------------------------------- /data_dir/hall_set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/data_dir/hall_set.py -------------------------------------------------------------------------------- /data_dir/process_ppg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/data_dir/process_ppg.py -------------------------------------------------------------------------------- /data_dir/process_uea.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/data_dir/process_uea.py -------------------------------------------------------------------------------- /data_dir/toy_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/data_dir/toy_dataset.py -------------------------------------------------------------------------------- /experiment_configs/repeats/S5/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S5/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S5/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S5/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S5/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S5/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S5/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S5/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S5/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S5/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S5/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S5/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S5/ppg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S5/ppg.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S6/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S6/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S6/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S6/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S6/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S6/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S6/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S6/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S6/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S6/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S6/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S6/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/S6/ppg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/S6/ppg.json -------------------------------------------------------------------------------- /experiment_configs/repeats/bd_linear_ncde/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/bd_linear_ncde/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/bd_linear_ncde/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/bd_linear_ncde/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/bd_linear_ncde/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/bd_linear_ncde/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/bd_linear_ncde/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/bd_linear_ncde/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/bd_linear_ncde/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/bd_linear_ncde/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/bd_linear_ncde/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/bd_linear_ncde/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dense_linear_ncde/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dense_linear_ncde/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dense_linear_ncde/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dense_linear_ncde/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dense_linear_ncde/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dense_linear_ncde/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dense_linear_ncde/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dense_linear_ncde/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dense_linear_ncde/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dense_linear_ncde/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dense_linear_ncde/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dense_linear_ncde/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_dense_linear_ncde/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_dense_linear_ncde/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_dense_linear_ncde/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_dense_linear_ncde/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_dense_linear_ncde/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_dense_linear_ncde/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_dense_linear_ncde/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_dense_linear_ncde/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_dense_linear_ncde/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_dense_linear_ncde/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_dense_linear_ncde/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_dense_linear_ncde/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_linear_ncde/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_linear_ncde/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_linear_ncde/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_linear_ncde/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_linear_ncde/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_linear_ncde/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_linear_ncde/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_linear_ncde/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_linear_ncde/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_linear_ncde/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/diagonal_linear_ncde/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/diagonal_linear_ncde/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dplr_linear_ncde/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dplr_linear_ncde/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dplr_linear_ncde/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dplr_linear_ncde/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dplr_linear_ncde/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dplr_linear_ncde/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dplr_linear_ncde/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dplr_linear_ncde/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dplr_linear_ncde/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dplr_linear_ncde/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/dplr_linear_ncde/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/dplr_linear_ncde/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/log_ncde/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/log_ncde/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/log_ncde/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/log_ncde/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/log_ncde/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/log_ncde/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/log_ncde/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/log_ncde/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/log_ncde/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/log_ncde/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/log_ncde/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/log_ncde/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/log_ncde/ppg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/log_ncde/ppg.json -------------------------------------------------------------------------------- /experiment_configs/repeats/lru/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/lru/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/lru/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/lru/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/lru/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/lru/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/lru/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/lru/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/lru/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/lru/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/lru/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/lru/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/lru/ppg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/lru/ppg.json -------------------------------------------------------------------------------- /experiment_configs/repeats/mamba/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/mamba/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/mamba/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/mamba/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/mamba/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/mamba/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/mamba/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/mamba/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/mamba/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/mamba/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/mamba/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/mamba/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/mamba/ppg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/mamba/ppg.json -------------------------------------------------------------------------------- /experiment_configs/repeats/ncde/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/ncde/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/ncde/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/ncde/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/ncde/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/ncde/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/ncde/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/ncde/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/ncde/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/ncde/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/ncde/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/ncde/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/ncde/ppg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/ncde/ppg.json -------------------------------------------------------------------------------- /experiment_configs/repeats/nrde/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/nrde/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/nrde/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/nrde/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/nrde/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/nrde/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/nrde/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/nrde/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/nrde/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/nrde/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/nrde/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/nrde/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/nrde/ppg.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/nrde/ppg.json -------------------------------------------------------------------------------- /experiment_configs/repeats/sparse_linear_ncde/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/sparse_linear_ncde/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/sparse_linear_ncde/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/sparse_linear_ncde/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/sparse_linear_ncde/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/sparse_linear_ncde/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/sparse_linear_ncde/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/sparse_linear_ncde/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/sparse_linear_ncde/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/sparse_linear_ncde/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/sparse_linear_ncde/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/sparse_linear_ncde/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/repeats/wh_linear_ncde/EigenWorms.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/wh_linear_ncde/EigenWorms.json -------------------------------------------------------------------------------- /experiment_configs/repeats/wh_linear_ncde/EthanolConcentration.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/wh_linear_ncde/EthanolConcentration.json -------------------------------------------------------------------------------- /experiment_configs/repeats/wh_linear_ncde/Heartbeat.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/wh_linear_ncde/Heartbeat.json -------------------------------------------------------------------------------- /experiment_configs/repeats/wh_linear_ncde/MotorImagery.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/wh_linear_ncde/MotorImagery.json -------------------------------------------------------------------------------- /experiment_configs/repeats/wh_linear_ncde/SelfRegulationSCP1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/wh_linear_ncde/SelfRegulationSCP1.json -------------------------------------------------------------------------------- /experiment_configs/repeats/wh_linear_ncde/SelfRegulationSCP2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/repeats/wh_linear_ncde/SelfRegulationSCP2.json -------------------------------------------------------------------------------- /experiment_configs/toy/S5/signature1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/S5/signature1.json -------------------------------------------------------------------------------- /experiment_configs/toy/S5/signature2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/S5/signature2.json -------------------------------------------------------------------------------- /experiment_configs/toy/S5/signature3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/S5/signature3.json -------------------------------------------------------------------------------- /experiment_configs/toy/S5/signature4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/S5/signature4.json -------------------------------------------------------------------------------- /experiment_configs/toy/S6/signature1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/S6/signature1.json -------------------------------------------------------------------------------- /experiment_configs/toy/S6/signature2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/S6/signature2.json -------------------------------------------------------------------------------- /experiment_configs/toy/S6/signature3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/S6/signature3.json -------------------------------------------------------------------------------- /experiment_configs/toy/S6/signature4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/S6/signature4.json -------------------------------------------------------------------------------- /experiment_configs/toy/log_ncde/signature1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/log_ncde/signature1.json -------------------------------------------------------------------------------- /experiment_configs/toy/log_ncde/signature2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/log_ncde/signature2.json -------------------------------------------------------------------------------- /experiment_configs/toy/log_ncde/signature3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/log_ncde/signature3.json -------------------------------------------------------------------------------- /experiment_configs/toy/log_ncde/signature4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/log_ncde/signature4.json -------------------------------------------------------------------------------- /experiment_configs/toy/lru/signature1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/lru/signature1.json -------------------------------------------------------------------------------- /experiment_configs/toy/lru/signature2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/lru/signature2.json -------------------------------------------------------------------------------- /experiment_configs/toy/lru/signature3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/lru/signature3.json -------------------------------------------------------------------------------- /experiment_configs/toy/lru/signature4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/lru/signature4.json -------------------------------------------------------------------------------- /experiment_configs/toy/mamba/signature1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/mamba/signature1.json -------------------------------------------------------------------------------- /experiment_configs/toy/mamba/signature2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/mamba/signature2.json -------------------------------------------------------------------------------- /experiment_configs/toy/mamba/signature3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/mamba/signature3.json -------------------------------------------------------------------------------- /experiment_configs/toy/mamba/signature4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/mamba/signature4.json -------------------------------------------------------------------------------- /experiment_configs/toy/ncde/signature1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/ncde/signature1.json -------------------------------------------------------------------------------- /experiment_configs/toy/ncde/signature2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/ncde/signature2.json -------------------------------------------------------------------------------- /experiment_configs/toy/ncde/signature3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/ncde/signature3.json -------------------------------------------------------------------------------- /experiment_configs/toy/ncde/signature4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/ncde/signature4.json -------------------------------------------------------------------------------- /experiment_configs/toy/nrde/signature1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/nrde/signature1.json -------------------------------------------------------------------------------- /experiment_configs/toy/nrde/signature2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/nrde/signature2.json -------------------------------------------------------------------------------- /experiment_configs/toy/nrde/signature3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/nrde/signature3.json -------------------------------------------------------------------------------- /experiment_configs/toy/nrde/signature4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/experiment_configs/toy/nrde/signature4.json -------------------------------------------------------------------------------- /models/LRU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/models/LRU.py -------------------------------------------------------------------------------- /models/LinearNeuralCDEs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/models/LinearNeuralCDEs.py -------------------------------------------------------------------------------- /models/LogNeuralCDEs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/models/LogNeuralCDEs.py -------------------------------------------------------------------------------- /models/NeuralCDEs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/models/NeuralCDEs.py -------------------------------------------------------------------------------- /models/RNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/models/RNN.py -------------------------------------------------------------------------------- /models/S5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/models/S5.py -------------------------------------------------------------------------------- /models/generate_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/models/generate_model.py -------------------------------------------------------------------------------- /results/analyse_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/results/analyse_results.py -------------------------------------------------------------------------------- /results/hyperparam_info_for_table.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/results/hyperparam_info_for_table.json -------------------------------------------------------------------------------- /results/logsig_length.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/results/logsig_length.py -------------------------------------------------------------------------------- /results/memory_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/results/memory_time.py -------------------------------------------------------------------------------- /results/memory_time_results.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/results/memory_time_results.json -------------------------------------------------------------------------------- /results/paper_outputs.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/results/paper_outputs.zip -------------------------------------------------------------------------------- /results/plot_toy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/results/plot_toy.py -------------------------------------------------------------------------------- /results/produce_hypparam_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/results/produce_hypparam_table.py -------------------------------------------------------------------------------- /results/time_vs_acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/results/time_vs_acc.py -------------------------------------------------------------------------------- /run_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/run_experiment.py -------------------------------------------------------------------------------- /simple_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/simple_example.ipynb -------------------------------------------------------------------------------- /torch_experiments/jax_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/torch_experiments/jax_dataset.py -------------------------------------------------------------------------------- /torch_experiments/s6_recurrence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/torch_experiments/s6_recurrence.py -------------------------------------------------------------------------------- /torch_experiments/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/torch_experiments/train.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benjamin-Walker/log-neural-cdes/HEAD/train.py --------------------------------------------------------------------------------