tobias/entropice
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Unfiy the training scripts and add SHAP

Browse source
This commit is contained in:
Tobias Hölzer 2026-02-08 22:06:49 +01:00
parent 073502c51d
commit 3ce6b6e867
38 changed files with 5876 additions and 681 deletions
Download patch file Download diff file Expand all files Collapse all files

10
pyproject.toml
View file

@ -67,8 +67,10 @@ dependencies = [
"ruff>=0.14.11,<0.15",
"pandas-stubs>=2.3.3.251201,<3",
"pytest>=9.0.2,<10",
"autogluon-tabular[all,mitra]>=1.5.0",
"shap>=0.50.0,<0.51", "h5py>=3.15.1,<4",
"autogluon-tabular[all,mitra,realmlp,interpret,fastai,tabm,tabpfn,tabdpt,tabpfnmix,tabicl,skew,imodels]>=1.5.0",
"shap>=0.50.0,<0.51",
"h5py>=3.15.1,<4",
"pydantic>=2.12.5,<3",
]
[project.scripts]
@ -77,8 +79,8 @@ darts = "entropice.ingest.darts:cli"
alpha-earth = "entropice.ingest.alphaearth:main"
era5 = "entropice.ingest.era5:cli"
arcticdem = "entropice.ingest.arcticdem:cli"
train = "entropice.ml.training:cli"
autogluon = "entropice.ml.autogluon_training:cli"
train = "entropice.ml.hpsearchcv:cli"
autogluon = "entropice.ml.autogluon:cli"
[build-system]
requires = ["hatchling"]