Unfiy the training scripts and add SHAP

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

View file

@ -29,7 +29,7 @@ import shutil
import pytest
from entropice.ml.dataset import DatasetEnsemble
from entropice.ml.training import CVSettings, random_cv
from entropice.ml.randomsearch import RunSettings, random_cv
from entropice.utils.types import Model, Task
@ -90,7 +90,7 @@ class TestRandomCV:
- All output files are created
"""
# Use darts_v1 as the primary target for all tests
settings = CVSettings(
settings = RunSettings(
n_iter=3,
task=task,
target="darts_v1",
@ -141,7 +141,7 @@ class TestRandomCV:
- xgboost: Uses CUDA without array API dispatch
- rf/knn: GPU-accelerated via cuML
"""
settings = CVSettings(
settings = RunSettings(
n_iter=3,
task="binary", # Simple binary task for device testing
target="darts_v1",
@ -168,7 +168,7 @@ class TestRandomCV:
def test_random_cv_with_mllabels(self, test_ensemble, cleanup_results):
"""Test random_cv with multi-label target dataset."""
settings = CVSettings(
settings = RunSettings(
n_iter=3,
task="binary",
target="darts_mllabels",
@ -199,7 +199,7 @@ class TestRandomCV:
add_lonlat=True,
)
settings = CVSettings(
settings = RunSettings(
n_iter=3,
task="binary",
target="darts_v1",