├── AI_ETHICS.md ├── CODEOWNERS ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING-ARCHIVED.md ├── CONTRIBUTING.md ├── LICENSE.txt ├── README.md ├── SECURITY.md ├── benchmark ├── __init__.py ├── benchmark_exp.py ├── callbacks │ ├── __init__.py │ └── pytorch_lightning.py ├── conf │ └── benchmark_exp.yaml ├── model │ ├── __init__.py │ ├── multivariate_stats_predictors.py │ └── torch │ │ ├── __init__.py │ │ ├── autoformer │ │ ├── __init__.py │ │ ├── estimator.py │ │ ├── layers.py │ │ ├── lightning_module.py │ │ └── module.py │ │ ├── deeptime │ │ ├── __init__.py │ │ ├── estimator.py │ │ ├── layers.py │ │ ├── lightning_module.py │ │ └── module.py │ │ ├── deepvar │ │ ├── __init__.py │ │ ├── estimator.py │ │ ├── lightning_module.py │ │ └── module.py │ │ ├── distributions.py │ │ ├── fedformer │ │ ├── __init__.py │ │ ├── estimator.py │ │ ├── layers.py │ │ ├── lightning_module.py │ │ └── module.py │ │ ├── linear_family │ │ ├── __init__.py │ │ ├── estimator.py │ │ ├── layers.py │ │ ├── lightning_module.py │ │ └── module.py │ │ ├── n_beats │ │ ├── __init__.py │ │ ├── layers.py │ │ ├── lightning_module.py │ │ └── module.py │ │ ├── ns_transformer │ │ ├── __init__.py │ │ ├── estimator.py │ │ ├── layers.py │ │ ├── lightning_module.py │ │ └── module.py │ │ ├── patch_tst │ │ ├── __init__.py │ │ ├── estimator.py │ │ ├── layers.py │ │ ├── lightning_module.py │ │ └── module.py │ │ ├── tft │ │ ├── __init__.py │ │ ├── estimator.py │ │ ├── layers.py │ │ ├── lightning_module.py │ │ └── module.py │ │ ├── timegrad │ │ ├── __init__.py │ │ ├── epsilon_theta.py │ │ ├── estimator.py │ │ ├── gaussian_diffusion.py │ │ ├── lightning_module.py │ │ └── module.py │ │ └── utils.py └── stats_exp.py ├── pretraining ├── __init__.py ├── conf │ ├── backbone │ │ ├── cost.yaml │ │ ├── encoder_cls.yaml │ │ ├── encoder_decoder.yaml │ │ ├── encoder_decoder_dms.yaml │ │ ├── encoder_flatten.yaml │ │ ├── encoder_mean.yaml │ │ ├── masked_encoder.yaml │ │ ├── meta_nbeats.yaml │ │ ├── one_fits_all.yaml │ │ └── ts2vec.yaml │ ├── forecast │ │ ├── finetune.yaml │ │ ├── meta_nbeats.yaml │ │ └── zeroshot.yaml │ ├── forecast_exp.yaml │ ├── pretrain │ │ ├── meta_nbeats.yaml │ │ └── supervised.yaml │ ├── pretrain_exp.yaml │ └── size │ │ ├── base.yaml │ │ ├── large.yaml │ │ ├── none.yaml │ │ └── xlarge.yaml ├── forecast_exp.py ├── model │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ ├── cost.py │ │ ├── encoder.py │ │ ├── encoder_decoder.py │ │ ├── encoder_decoder_dms.py │ │ ├── layers │ │ │ ├── __init__.py │ │ │ ├── attention.py │ │ │ ├── embeddings.py │ │ │ ├── mlp.py │ │ │ └── transformer.py │ │ ├── masked_encoder.py │ │ ├── meta_nbeats.py │ │ ├── one_fits_all.py │ │ └── ts2vec.py │ ├── forecast │ │ ├── __init__.py │ │ ├── finetune.py │ │ ├── meta_nbeats.py │ │ └── zeroshot.py │ └── pretrain │ │ ├── __init__.py │ │ ├── meta_nbeats.py │ │ └── supervised.py └── pretrain_exp.py ├── requirements ├── requirements-pytorch.txt └── requirements-stats.txt └── util ├── __init__.py ├── dataset.py ├── evaluation.py ├── forecast_generator.py ├── prepare.py ├── torch ├── __init__.py ├── attn_mask.py ├── distributions │ ├── __init__.py │ ├── multivariate_studentT.py │ ├── normalizing_flow.py │ └── spline_quantile_function.py ├── lightning_module.py ├── ops.py ├── quantile_output.py └── scaler.py └── transform ├── __init__.py ├── convert.py ├── sampler.py └── split.py /AI_ETHICS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/AI_ETHICS.md -------------------------------------------------------------------------------- /CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/CODEOWNERS -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING-ARCHIVED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/CONTRIBUTING-ARCHIVED.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/SECURITY.md -------------------------------------------------------------------------------- /benchmark/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmark/benchmark_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/benchmark_exp.py -------------------------------------------------------------------------------- /benchmark/callbacks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmark/callbacks/pytorch_lightning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/callbacks/pytorch_lightning.py -------------------------------------------------------------------------------- /benchmark/conf/benchmark_exp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/conf/benchmark_exp.yaml -------------------------------------------------------------------------------- /benchmark/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmark/model/multivariate_stats_predictors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/multivariate_stats_predictors.py -------------------------------------------------------------------------------- /benchmark/model/torch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/__init__.py -------------------------------------------------------------------------------- /benchmark/model/torch/autoformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/autoformer/__init__.py -------------------------------------------------------------------------------- /benchmark/model/torch/autoformer/estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/autoformer/estimator.py -------------------------------------------------------------------------------- /benchmark/model/torch/autoformer/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/autoformer/layers.py -------------------------------------------------------------------------------- /benchmark/model/torch/autoformer/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/autoformer/lightning_module.py -------------------------------------------------------------------------------- /benchmark/model/torch/autoformer/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/autoformer/module.py -------------------------------------------------------------------------------- /benchmark/model/torch/deeptime/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/deeptime/__init__.py -------------------------------------------------------------------------------- /benchmark/model/torch/deeptime/estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/deeptime/estimator.py -------------------------------------------------------------------------------- /benchmark/model/torch/deeptime/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/deeptime/layers.py -------------------------------------------------------------------------------- /benchmark/model/torch/deeptime/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/deeptime/lightning_module.py -------------------------------------------------------------------------------- /benchmark/model/torch/deeptime/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/deeptime/module.py -------------------------------------------------------------------------------- /benchmark/model/torch/deepvar/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/deepvar/__init__.py -------------------------------------------------------------------------------- /benchmark/model/torch/deepvar/estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/deepvar/estimator.py -------------------------------------------------------------------------------- /benchmark/model/torch/deepvar/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/deepvar/lightning_module.py -------------------------------------------------------------------------------- /benchmark/model/torch/deepvar/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/deepvar/module.py -------------------------------------------------------------------------------- /benchmark/model/torch/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/distributions.py -------------------------------------------------------------------------------- /benchmark/model/torch/fedformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/fedformer/__init__.py -------------------------------------------------------------------------------- /benchmark/model/torch/fedformer/estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/fedformer/estimator.py -------------------------------------------------------------------------------- /benchmark/model/torch/fedformer/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/fedformer/layers.py -------------------------------------------------------------------------------- /benchmark/model/torch/fedformer/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/fedformer/lightning_module.py -------------------------------------------------------------------------------- /benchmark/model/torch/fedformer/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/fedformer/module.py -------------------------------------------------------------------------------- /benchmark/model/torch/linear_family/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/linear_family/__init__.py -------------------------------------------------------------------------------- /benchmark/model/torch/linear_family/estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/linear_family/estimator.py -------------------------------------------------------------------------------- /benchmark/model/torch/linear_family/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/linear_family/layers.py -------------------------------------------------------------------------------- /benchmark/model/torch/linear_family/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/linear_family/lightning_module.py -------------------------------------------------------------------------------- /benchmark/model/torch/linear_family/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/linear_family/module.py -------------------------------------------------------------------------------- /benchmark/model/torch/n_beats/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/n_beats/__init__.py -------------------------------------------------------------------------------- /benchmark/model/torch/n_beats/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/n_beats/layers.py -------------------------------------------------------------------------------- /benchmark/model/torch/n_beats/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/n_beats/lightning_module.py -------------------------------------------------------------------------------- /benchmark/model/torch/n_beats/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/n_beats/module.py -------------------------------------------------------------------------------- /benchmark/model/torch/ns_transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/ns_transformer/__init__.py -------------------------------------------------------------------------------- /benchmark/model/torch/ns_transformer/estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/ns_transformer/estimator.py -------------------------------------------------------------------------------- /benchmark/model/torch/ns_transformer/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/ns_transformer/layers.py -------------------------------------------------------------------------------- /benchmark/model/torch/ns_transformer/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/ns_transformer/lightning_module.py -------------------------------------------------------------------------------- /benchmark/model/torch/ns_transformer/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/ns_transformer/module.py -------------------------------------------------------------------------------- /benchmark/model/torch/patch_tst/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/patch_tst/__init__.py -------------------------------------------------------------------------------- /benchmark/model/torch/patch_tst/estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/patch_tst/estimator.py -------------------------------------------------------------------------------- /benchmark/model/torch/patch_tst/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/patch_tst/layers.py -------------------------------------------------------------------------------- /benchmark/model/torch/patch_tst/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/patch_tst/lightning_module.py -------------------------------------------------------------------------------- /benchmark/model/torch/patch_tst/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/patch_tst/module.py -------------------------------------------------------------------------------- /benchmark/model/torch/tft/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/tft/__init__.py -------------------------------------------------------------------------------- /benchmark/model/torch/tft/estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/tft/estimator.py -------------------------------------------------------------------------------- /benchmark/model/torch/tft/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/tft/layers.py -------------------------------------------------------------------------------- /benchmark/model/torch/tft/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/tft/lightning_module.py -------------------------------------------------------------------------------- /benchmark/model/torch/tft/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/tft/module.py -------------------------------------------------------------------------------- /benchmark/model/torch/timegrad/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/timegrad/__init__.py -------------------------------------------------------------------------------- /benchmark/model/torch/timegrad/epsilon_theta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/timegrad/epsilon_theta.py -------------------------------------------------------------------------------- /benchmark/model/torch/timegrad/estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/timegrad/estimator.py -------------------------------------------------------------------------------- /benchmark/model/torch/timegrad/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/timegrad/gaussian_diffusion.py -------------------------------------------------------------------------------- /benchmark/model/torch/timegrad/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/timegrad/lightning_module.py -------------------------------------------------------------------------------- /benchmark/model/torch/timegrad/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/timegrad/module.py -------------------------------------------------------------------------------- /benchmark/model/torch/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/model/torch/utils.py -------------------------------------------------------------------------------- /benchmark/stats_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/benchmark/stats_exp.py -------------------------------------------------------------------------------- /pretraining/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretraining/conf/backbone/cost.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretraining/conf/backbone/encoder_cls.yaml: -------------------------------------------------------------------------------- 1 | decoder_mode: cls -------------------------------------------------------------------------------- /pretraining/conf/backbone/encoder_decoder.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretraining/conf/backbone/encoder_decoder_dms.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretraining/conf/backbone/encoder_flatten.yaml: -------------------------------------------------------------------------------- 1 | decoder_mode: flatten -------------------------------------------------------------------------------- /pretraining/conf/backbone/encoder_mean.yaml: -------------------------------------------------------------------------------- 1 | decoder_mode: mean -------------------------------------------------------------------------------- /pretraining/conf/backbone/masked_encoder.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretraining/conf/backbone/meta_nbeats.yaml: -------------------------------------------------------------------------------- 1 | model_type: generic -------------------------------------------------------------------------------- /pretraining/conf/backbone/one_fits_all.yaml: -------------------------------------------------------------------------------- 1 | patch_length: 16 2 | stride: 8 -------------------------------------------------------------------------------- /pretraining/conf/backbone/ts2vec.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretraining/conf/forecast/finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/conf/forecast/finetune.yaml -------------------------------------------------------------------------------- /pretraining/conf/forecast/meta_nbeats.yaml: -------------------------------------------------------------------------------- 1 | loss: smape 2 | context_len_mult: 11 3 | lr: 1e-3 4 | weight_decay: 0 -------------------------------------------------------------------------------- /pretraining/conf/forecast/zeroshot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/conf/forecast/zeroshot.yaml -------------------------------------------------------------------------------- /pretraining/conf/forecast_exp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/conf/forecast_exp.yaml -------------------------------------------------------------------------------- /pretraining/conf/pretrain/meta_nbeats.yaml: -------------------------------------------------------------------------------- 1 | loss: smape 2 | context_len_mult: 11 3 | lr: 1e-3 4 | weight_decay: 0 -------------------------------------------------------------------------------- /pretraining/conf/pretrain/supervised.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/conf/pretrain/supervised.yaml -------------------------------------------------------------------------------- /pretraining/conf/pretrain_exp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/conf/pretrain_exp.yaml -------------------------------------------------------------------------------- /pretraining/conf/size/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/conf/size/base.yaml -------------------------------------------------------------------------------- /pretraining/conf/size/large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/conf/size/large.yaml -------------------------------------------------------------------------------- /pretraining/conf/size/none.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretraining/conf/size/xlarge.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/conf/size/xlarge.yaml -------------------------------------------------------------------------------- /pretraining/forecast_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/forecast_exp.py -------------------------------------------------------------------------------- /pretraining/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretraining/model/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/__init__.py -------------------------------------------------------------------------------- /pretraining/model/backbone/cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/cost.py -------------------------------------------------------------------------------- /pretraining/model/backbone/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/encoder.py -------------------------------------------------------------------------------- /pretraining/model/backbone/encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/encoder_decoder.py -------------------------------------------------------------------------------- /pretraining/model/backbone/encoder_decoder_dms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/encoder_decoder_dms.py -------------------------------------------------------------------------------- /pretraining/model/backbone/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretraining/model/backbone/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/layers/attention.py -------------------------------------------------------------------------------- /pretraining/model/backbone/layers/embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/layers/embeddings.py -------------------------------------------------------------------------------- /pretraining/model/backbone/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/layers/mlp.py -------------------------------------------------------------------------------- /pretraining/model/backbone/layers/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/layers/transformer.py -------------------------------------------------------------------------------- /pretraining/model/backbone/masked_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/masked_encoder.py -------------------------------------------------------------------------------- /pretraining/model/backbone/meta_nbeats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/meta_nbeats.py -------------------------------------------------------------------------------- /pretraining/model/backbone/one_fits_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/one_fits_all.py -------------------------------------------------------------------------------- /pretraining/model/backbone/ts2vec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/backbone/ts2vec.py -------------------------------------------------------------------------------- /pretraining/model/forecast/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/forecast/__init__.py -------------------------------------------------------------------------------- /pretraining/model/forecast/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/forecast/finetune.py -------------------------------------------------------------------------------- /pretraining/model/forecast/meta_nbeats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/forecast/meta_nbeats.py -------------------------------------------------------------------------------- /pretraining/model/forecast/zeroshot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/forecast/zeroshot.py -------------------------------------------------------------------------------- /pretraining/model/pretrain/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/pretrain/__init__.py -------------------------------------------------------------------------------- /pretraining/model/pretrain/meta_nbeats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/pretrain/meta_nbeats.py -------------------------------------------------------------------------------- /pretraining/model/pretrain/supervised.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/model/pretrain/supervised.py -------------------------------------------------------------------------------- /pretraining/pretrain_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/pretraining/pretrain_exp.py -------------------------------------------------------------------------------- /requirements/requirements-pytorch.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/requirements/requirements-pytorch.txt -------------------------------------------------------------------------------- /requirements/requirements-stats.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/requirements/requirements-stats.txt -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /util/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/dataset.py -------------------------------------------------------------------------------- /util/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/evaluation.py -------------------------------------------------------------------------------- /util/forecast_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/forecast_generator.py -------------------------------------------------------------------------------- /util/prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/prepare.py -------------------------------------------------------------------------------- /util/torch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /util/torch/attn_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/torch/attn_mask.py -------------------------------------------------------------------------------- /util/torch/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/torch/distributions/__init__.py -------------------------------------------------------------------------------- /util/torch/distributions/multivariate_studentT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/torch/distributions/multivariate_studentT.py -------------------------------------------------------------------------------- /util/torch/distributions/normalizing_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/torch/distributions/normalizing_flow.py -------------------------------------------------------------------------------- /util/torch/distributions/spline_quantile_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/torch/distributions/spline_quantile_function.py -------------------------------------------------------------------------------- /util/torch/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/torch/lightning_module.py -------------------------------------------------------------------------------- /util/torch/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/torch/ops.py -------------------------------------------------------------------------------- /util/torch/quantile_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/torch/quantile_output.py -------------------------------------------------------------------------------- /util/torch/scaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/torch/scaler.py -------------------------------------------------------------------------------- /util/transform/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /util/transform/convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/transform/convert.py -------------------------------------------------------------------------------- /util/transform/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/transform/sampler.py -------------------------------------------------------------------------------- /util/transform/split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/pretrain-time-series-cloudops/HEAD/util/transform/split.py --------------------------------------------------------------------------------