Pretrained models for "Stepwise Benchmarking: Data-Scarce Groundwater Heat Pumps' Modeling"
This dataset contains the pretrained baseline models, namely CNO, ConvNeXt U-Net, DPOT, LGCNN, scOT, TFNO, and U-Net. A model is trained under a fixed budget on all three training steps in the benchmark and included if training is technically possible under these constraints.
The U-Net and LGCNN models are stored in .pt files, the default file type for the PyTorch library, with a configuration file in .yaml format containing the hyperparameters. The configuration file can be used to automatically instantiate the correct model with our Github repository: https://github.com/JuliaPelzer/Heat-Plume-Prediction/tree/AllIn1/LGCNN/benchmarkingdata (commit: 83bf879).
More details can be found in the implementation.
The training code for the other models is available at https://github.com/kotserge/benchmarking-heat-pump (commit: 1eb5e7a).
Automatic evaluation of all models can be obtained with https://github.com/JuliaPelzer/Stepwise-Benchmarking-HPs.
SWE models are trained on the data from Takamoto, Makoto; Praditia, Timothy; Leiteritz, Raphael; MacKinlay, Dan; Alesiani, Francesco; Pflüger, Dirk; Niepert, Mathias, 2022, "PDEBench Datasets",
https://doi.org/10.18419/DARUS-2986, DaRUS, V8.