diff --git a/configs/regular/logreg_cv.json b/configs/regular/logreg_cv.json new file mode 100644 index 00000000..ec7a6a6d --- /dev/null +++ b/configs/regular/logreg_cv.json @@ -0,0 +1,47 @@ +{ + "INCLUDE": ["../common/sklearn.json"], + "PARAMETERS_SETS": { + "sklearn logreg_cv parameters": { + "algorithm": { + "estimator": "LogisticRegressionCV", + "library": ["sklearn", "sklearnex"], + "device": "cpu", + "estimator_methods": { "inference": "predict" }, + "estimator_params": { + "Cs": 10, + "solver": "lbfgs", + "max_iter": 1000, + "random_state": 123 + } + } + }, + "logreg_cv datasets": [ + { + "data": { + "source": "make_classification", + "generation_kwargs": [ + { + "n_samples": 100000, + "n_features": 1000, + "n_informative": 800, + "n_classes": [2] + } + ], + "split_kwargs": { + "train_size": 0.05, + "test_size": 0.95 + } + } + } + ] + }, + "TEMPLATES": { + "sklearn logreg_cv": { + "SETS": [ + "sklearn-ex[cpu,gpu] implementations", + "sklearn logreg_cv parameters", + "logreg_cv datasets" + ] + } + } +} diff --git a/envs/conda-env-sklearn.yml b/envs/conda-env-sklearn.yml index d8f0571f..3dcbabc6 100644 --- a/envs/conda-env-sklearn.yml +++ b/envs/conda-env-sklearn.yml @@ -10,7 +10,7 @@ dependencies: - modin-all - scikit-learn-intelex # sklbench dependencies - - scikit-learn + - scikit-learn>=1.8 - pandas - tabulate - fastparquet diff --git a/envs/requirements-sklearn.txt b/envs/requirements-sklearn.txt index 064c9e90..6067dce2 100644 --- a/envs/requirements-sklearn.txt +++ b/envs/requirements-sklearn.txt @@ -8,7 +8,7 @@ scikit-learn-intelex dpctl dpnp # sklbench dependencies -scikit-learn +scikit-learn>=1.8 pandas tabulate fastparquet diff --git a/sklbench/benchmarks/estimator_task_map.json b/sklbench/benchmarks/estimator_task_map.json index 09eb59b5..d5a120f2 100644 --- a/sklbench/benchmarks/estimator_task_map.json +++ b/sklbench/benchmarks/estimator_task_map.json @@ -2,6 +2,7 @@ "classification": [ "Classifier", "LogisticRegression", + "LogisticRegressionCV", "SVC" ], "regression": [