Unfiy the training scripts and add SHAP
This commit is contained in:
parent
073502c51d
commit
3ce6b6e867
38 changed files with 5876 additions and 681 deletions
|
|
@ -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",
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue