├── .editorconfig ├── .github ├── ISSUE_TEMPLATE │ ├── bug_report.md │ ├── config.yml │ └── feature_request.md └── workflows │ ├── build.yml │ ├── docs.yml │ └── pypi.yml ├── .gitignore ├── .python-version ├── AUTHORS.rst ├── LICENSE ├── MANIFEST.in ├── README.md ├── README.zh.md ├── data ├── basin_5819.csv ├── basin_615.csv ├── basin_id(0+20).csv ├── basin_id(46+1).csv ├── basin_id(498+24).csv ├── basin_id(498+41).csv ├── basin_id(97+24).csv ├── basin_id(all).csv ├── gage_id.csv └── gage_id_test.csv ├── docs ├── api │ ├── configs.md │ ├── datasets.md │ ├── explainers.md │ ├── models.md │ └── trainers.md ├── authors.rst ├── changelog.md ├── contributing.md ├── fabric_debug_guide.md ├── faq.md ├── index.md ├── installation.md ├── overrides │ └── main.html ├── torchhydro.md └── usage.md ├── examples ├── dpl_xaj_example.py └── lstm_camels_example.py ├── experiments ├── .DS_Store ├── camels531.csv ├── check_model_state.py ├── evaluate_with_camels_17attributes.py ├── evaluate_with_camels_sk.py ├── evaluate_with_camels_sk_aef.py ├── evaluate_with_era5land.py ├── evaluate_with_gpm.py ├── evaluate_with_gpm_streamflow.py ├── evaluate_with_mtslstm.py ├── evaluate_with_us_aef.py ├── generate_pt_and_test_loaded.py ├── train_lstm_with_sanxia.py ├── train_mtslstm_with_sanxia.py ├── train_with_aus.py ├── train_with_camels_17attributes.py ├── train_with_camels_sk.py ├── train_with_camelssk_aef.py ├── train_with_era5land.py ├── train_with_era5land_gnn_ddp.py ├── train_with_era5land_trans.py ├── train_with_gpm.py ├── train_with_gpm_dis.py ├── train_with_gpm_streamflow.py ├── train_with_mtslstm.py ├── train_with_songliaorrevent_gnn.py └── train_with_us_aef.py ├── mkdocs.yml ├── pyproject.toml ├── tests ├── __init__.py ├── conftest.py ├── test_ann.py ├── test_caravan_train.py ├── test_config.py ├── test_cudnnlstm.py ├── test_data_scalers.py ├── test_data_sets.py ├── test_data_utils.py ├── test_dataloader.py ├── test_deep_hydro.py ├── test_dpl4hbv.py ├── test_dpl4xaj.py ├── test_dropout.py ├── test_ensemble_exps.py ├── test_evaluate_grid_lstm.py ├── test_evaluate_model.py ├── test_federated_learning.py ├── test_flood_event_datasource.py ├── test_model_utils.py ├── test_mtslstm.py ├── test_resulter.py ├── test_sampler.py ├── test_selfmadehydrodataset_models.py ├── test_seq2seq.py ├── test_seqforecast.py ├── test_tl_selfmadedata.py ├── test_train_camels_lstm.py ├── test_train_camelspro_mtl.py ├── test_train_grid_lstm.py ├── test_train_utils.py ├── test_transfer_learning.py └── test_weight_analysis.py ├── torchhydro ├── .DS_Store ├── __init__.py ├── configs │ ├── __init__.py │ ├── config.py │ └── model_config.py ├── datasets │ ├── __init__.py │ ├── data_dict.py │ ├── data_scalers.py │ ├── data_sets.py │ ├── data_sources.py │ ├── data_utils.py │ └── sampler.py ├── explainers │ ├── __init__.py │ ├── shap.py │ ├── uncertainty_analysis.py │ └── weight_anlysis.py ├── models │ ├── __init__.py │ ├── ann.py │ ├── crits.py │ ├── cudnnlstm.py │ ├── dpl4gr4j.py │ ├── dpl4hbv.py │ ├── dpl4xaj.py │ ├── dpl4xaj_nn4et.py │ ├── dropout.py │ ├── gnn.py │ ├── kernel_conv.py │ ├── model_dict_function.py │ ├── model_utils.py │ ├── mtslstm.py │ ├── seq2seq.py │ ├── seqforecast.py │ ├── simple_lstm.py │ └── spplstm.py └── trainers │ ├── __init__.py │ ├── deep_hydro.py │ ├── fabric_wrapper.py │ ├── resulter.py │ ├── train_logger.py │ ├── train_utils.py │ └── trainer.py └── uv.lock /.editorconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/.editorconfig -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/workflows/build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/.github/workflows/build.yml -------------------------------------------------------------------------------- /.github/workflows/docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/.github/workflows/docs.yml -------------------------------------------------------------------------------- /.github/workflows/pypi.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/.github/workflows/pypi.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/.gitignore -------------------------------------------------------------------------------- /.python-version: -------------------------------------------------------------------------------- 1 | 3.11 -------------------------------------------------------------------------------- /AUTHORS.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/AUTHORS.rst -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/README.md -------------------------------------------------------------------------------- /README.zh.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/README.zh.md -------------------------------------------------------------------------------- /data/basin_5819.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/data/basin_5819.csv -------------------------------------------------------------------------------- /data/basin_615.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/data/basin_615.csv -------------------------------------------------------------------------------- /data/basin_id(0+20).csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/data/basin_id(0+20).csv -------------------------------------------------------------------------------- /data/basin_id(46+1).csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/data/basin_id(46+1).csv -------------------------------------------------------------------------------- /data/basin_id(498+24).csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/data/basin_id(498+24).csv -------------------------------------------------------------------------------- /data/basin_id(498+41).csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/data/basin_id(498+41).csv -------------------------------------------------------------------------------- /data/basin_id(97+24).csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/data/basin_id(97+24).csv -------------------------------------------------------------------------------- /data/basin_id(all).csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/data/basin_id(all).csv -------------------------------------------------------------------------------- /data/gage_id.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/data/gage_id.csv -------------------------------------------------------------------------------- /data/gage_id_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/data/gage_id_test.csv -------------------------------------------------------------------------------- /docs/api/configs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/api/configs.md -------------------------------------------------------------------------------- /docs/api/datasets.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/api/datasets.md -------------------------------------------------------------------------------- /docs/api/explainers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/api/explainers.md -------------------------------------------------------------------------------- /docs/api/models.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/api/models.md -------------------------------------------------------------------------------- /docs/api/trainers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/api/trainers.md -------------------------------------------------------------------------------- /docs/authors.rst: -------------------------------------------------------------------------------- 1 | .. include:: ../AUTHORS.rst 2 | -------------------------------------------------------------------------------- /docs/changelog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/changelog.md -------------------------------------------------------------------------------- /docs/contributing.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/contributing.md -------------------------------------------------------------------------------- /docs/fabric_debug_guide.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/fabric_debug_guide.md -------------------------------------------------------------------------------- /docs/faq.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/faq.md -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- 1 | {% include-markdown "../README.md" %} 2 | -------------------------------------------------------------------------------- /docs/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/installation.md -------------------------------------------------------------------------------- /docs/overrides/main.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/overrides/main.html -------------------------------------------------------------------------------- /docs/torchhydro.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/torchhydro.md -------------------------------------------------------------------------------- /docs/usage.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/docs/usage.md -------------------------------------------------------------------------------- /examples/dpl_xaj_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/examples/dpl_xaj_example.py -------------------------------------------------------------------------------- /examples/lstm_camels_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/examples/lstm_camels_example.py -------------------------------------------------------------------------------- /experiments/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/.DS_Store -------------------------------------------------------------------------------- /experiments/camels531.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/camels531.csv -------------------------------------------------------------------------------- /experiments/check_model_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/check_model_state.py -------------------------------------------------------------------------------- /experiments/evaluate_with_camels_17attributes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/evaluate_with_camels_17attributes.py -------------------------------------------------------------------------------- /experiments/evaluate_with_camels_sk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/evaluate_with_camels_sk.py -------------------------------------------------------------------------------- /experiments/evaluate_with_camels_sk_aef.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/evaluate_with_camels_sk_aef.py -------------------------------------------------------------------------------- /experiments/evaluate_with_era5land.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/evaluate_with_era5land.py -------------------------------------------------------------------------------- /experiments/evaluate_with_gpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/evaluate_with_gpm.py -------------------------------------------------------------------------------- /experiments/evaluate_with_gpm_streamflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/evaluate_with_gpm_streamflow.py -------------------------------------------------------------------------------- /experiments/evaluate_with_mtslstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/evaluate_with_mtslstm.py -------------------------------------------------------------------------------- /experiments/evaluate_with_us_aef.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/evaluate_with_us_aef.py -------------------------------------------------------------------------------- /experiments/generate_pt_and_test_loaded.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/generate_pt_and_test_loaded.py -------------------------------------------------------------------------------- /experiments/train_lstm_with_sanxia.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_lstm_with_sanxia.py -------------------------------------------------------------------------------- /experiments/train_mtslstm_with_sanxia.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_mtslstm_with_sanxia.py -------------------------------------------------------------------------------- /experiments/train_with_aus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_aus.py -------------------------------------------------------------------------------- /experiments/train_with_camels_17attributes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_camels_17attributes.py -------------------------------------------------------------------------------- /experiments/train_with_camels_sk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_camels_sk.py -------------------------------------------------------------------------------- /experiments/train_with_camelssk_aef.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_camelssk_aef.py -------------------------------------------------------------------------------- /experiments/train_with_era5land.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_era5land.py -------------------------------------------------------------------------------- /experiments/train_with_era5land_gnn_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_era5land_gnn_ddp.py -------------------------------------------------------------------------------- /experiments/train_with_era5land_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_era5land_trans.py -------------------------------------------------------------------------------- /experiments/train_with_gpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_gpm.py -------------------------------------------------------------------------------- /experiments/train_with_gpm_dis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_gpm_dis.py -------------------------------------------------------------------------------- /experiments/train_with_gpm_streamflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_gpm_streamflow.py -------------------------------------------------------------------------------- /experiments/train_with_mtslstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_mtslstm.py -------------------------------------------------------------------------------- /experiments/train_with_songliaorrevent_gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_songliaorrevent_gnn.py -------------------------------------------------------------------------------- /experiments/train_with_us_aef.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/experiments/train_with_us_aef.py -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/test_ann.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_ann.py -------------------------------------------------------------------------------- /tests/test_caravan_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_caravan_train.py -------------------------------------------------------------------------------- /tests/test_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_config.py -------------------------------------------------------------------------------- /tests/test_cudnnlstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_cudnnlstm.py -------------------------------------------------------------------------------- /tests/test_data_scalers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_data_scalers.py -------------------------------------------------------------------------------- /tests/test_data_sets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_data_sets.py -------------------------------------------------------------------------------- /tests/test_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_data_utils.py -------------------------------------------------------------------------------- /tests/test_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_dataloader.py -------------------------------------------------------------------------------- /tests/test_deep_hydro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_deep_hydro.py -------------------------------------------------------------------------------- /tests/test_dpl4hbv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_dpl4hbv.py -------------------------------------------------------------------------------- /tests/test_dpl4xaj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_dpl4xaj.py -------------------------------------------------------------------------------- /tests/test_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_dropout.py -------------------------------------------------------------------------------- /tests/test_ensemble_exps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_ensemble_exps.py -------------------------------------------------------------------------------- /tests/test_evaluate_grid_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_evaluate_grid_lstm.py -------------------------------------------------------------------------------- /tests/test_evaluate_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_evaluate_model.py -------------------------------------------------------------------------------- /tests/test_federated_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_federated_learning.py -------------------------------------------------------------------------------- /tests/test_flood_event_datasource.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_flood_event_datasource.py -------------------------------------------------------------------------------- /tests/test_model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_model_utils.py -------------------------------------------------------------------------------- /tests/test_mtslstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_mtslstm.py -------------------------------------------------------------------------------- /tests/test_resulter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_resulter.py -------------------------------------------------------------------------------- /tests/test_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_sampler.py -------------------------------------------------------------------------------- /tests/test_selfmadehydrodataset_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_selfmadehydrodataset_models.py -------------------------------------------------------------------------------- /tests/test_seq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_seq2seq.py -------------------------------------------------------------------------------- /tests/test_seqforecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_seqforecast.py -------------------------------------------------------------------------------- /tests/test_tl_selfmadedata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_tl_selfmadedata.py -------------------------------------------------------------------------------- /tests/test_train_camels_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_train_camels_lstm.py -------------------------------------------------------------------------------- /tests/test_train_camelspro_mtl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_train_camelspro_mtl.py -------------------------------------------------------------------------------- /tests/test_train_grid_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_train_grid_lstm.py -------------------------------------------------------------------------------- /tests/test_train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_train_utils.py -------------------------------------------------------------------------------- /tests/test_transfer_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_transfer_learning.py -------------------------------------------------------------------------------- /tests/test_weight_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/tests/test_weight_analysis.py -------------------------------------------------------------------------------- /torchhydro/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/.DS_Store -------------------------------------------------------------------------------- /torchhydro/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/__init__.py -------------------------------------------------------------------------------- /torchhydro/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchhydro/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/configs/config.py -------------------------------------------------------------------------------- /torchhydro/configs/model_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/configs/model_config.py -------------------------------------------------------------------------------- /torchhydro/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchhydro/datasets/data_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/datasets/data_dict.py -------------------------------------------------------------------------------- /torchhydro/datasets/data_scalers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/datasets/data_scalers.py -------------------------------------------------------------------------------- /torchhydro/datasets/data_sets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/datasets/data_sets.py -------------------------------------------------------------------------------- /torchhydro/datasets/data_sources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/datasets/data_sources.py -------------------------------------------------------------------------------- /torchhydro/datasets/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/datasets/data_utils.py -------------------------------------------------------------------------------- /torchhydro/datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/datasets/sampler.py -------------------------------------------------------------------------------- /torchhydro/explainers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchhydro/explainers/shap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/explainers/shap.py -------------------------------------------------------------------------------- /torchhydro/explainers/uncertainty_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/explainers/uncertainty_analysis.py -------------------------------------------------------------------------------- /torchhydro/explainers/weight_anlysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/explainers/weight_anlysis.py -------------------------------------------------------------------------------- /torchhydro/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/__init__.py -------------------------------------------------------------------------------- /torchhydro/models/ann.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/ann.py -------------------------------------------------------------------------------- /torchhydro/models/crits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/crits.py -------------------------------------------------------------------------------- /torchhydro/models/cudnnlstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/cudnnlstm.py -------------------------------------------------------------------------------- /torchhydro/models/dpl4gr4j.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/dpl4gr4j.py -------------------------------------------------------------------------------- /torchhydro/models/dpl4hbv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/dpl4hbv.py -------------------------------------------------------------------------------- /torchhydro/models/dpl4xaj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/dpl4xaj.py -------------------------------------------------------------------------------- /torchhydro/models/dpl4xaj_nn4et.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/dpl4xaj_nn4et.py -------------------------------------------------------------------------------- /torchhydro/models/dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/dropout.py -------------------------------------------------------------------------------- /torchhydro/models/gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/gnn.py -------------------------------------------------------------------------------- /torchhydro/models/kernel_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/kernel_conv.py -------------------------------------------------------------------------------- /torchhydro/models/model_dict_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/model_dict_function.py -------------------------------------------------------------------------------- /torchhydro/models/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/model_utils.py -------------------------------------------------------------------------------- /torchhydro/models/mtslstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/mtslstm.py -------------------------------------------------------------------------------- /torchhydro/models/seq2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/seq2seq.py -------------------------------------------------------------------------------- /torchhydro/models/seqforecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/seqforecast.py -------------------------------------------------------------------------------- /torchhydro/models/simple_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/simple_lstm.py -------------------------------------------------------------------------------- /torchhydro/models/spplstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/models/spplstm.py -------------------------------------------------------------------------------- /torchhydro/trainers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchhydro/trainers/deep_hydro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/trainers/deep_hydro.py -------------------------------------------------------------------------------- /torchhydro/trainers/fabric_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/trainers/fabric_wrapper.py -------------------------------------------------------------------------------- /torchhydro/trainers/resulter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/trainers/resulter.py -------------------------------------------------------------------------------- /torchhydro/trainers/train_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/trainers/train_logger.py -------------------------------------------------------------------------------- /torchhydro/trainers/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/trainers/train_utils.py -------------------------------------------------------------------------------- /torchhydro/trainers/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/torchhydro/trainers/trainer.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangWenyu/torchhydro/HEAD/uv.lock --------------------------------------------------------------------------------