From 00e43565dc0edf773acef4a51d27b0b887e86d5c Mon Sep 17 00:00:00 2001 From: ethanglaser Date: Thu, 23 Apr 2026 12:15:35 -0700 Subject: [PATCH 1/2] Improve sklearnex relative perf in sklearn_example.json --- configs/sklearn_example.json | 40 +++++++++++++++++++----------------- 1 file changed, 21 insertions(+), 19 deletions(-) diff --git a/configs/sklearn_example.json b/configs/sklearn_example.json index d746d1b2..76c7422c 100644 --- a/configs/sklearn_example.json +++ b/configs/sklearn_example.json @@ -13,7 +13,7 @@ "source": "make_blobs", "generation_kwargs": { "centers": 2, - "n_samples": 1000, + "n_samples": 5000, "n_features": [16, 64] }, "split_kwargs": { "ignore": true } @@ -33,29 +33,33 @@ } ] }, - "multi clsf data": { + "regression data": { "data": { - "source": "make_classification", + "source": "make_regression", "generation_kwargs": { - "n_classes": 4, - "n_samples": 1000, - "n_features": 10, - "n_informative": "[SPECIAL_VALUE]0.75" + "n_samples": 10000, + "n_features": 100, + "n_informative": 50, + "noise": 0.1 }, - "split_kwargs": { "test_size": 0.1 } + "split_kwargs": { "test_size": 0.2 } } }, + "regression algorithms": { + "algorithm": [ + { "estimator": "LinearRegression" } + ] + }, "unsupervised algorithms": { - "algorithm": - [ - { - "estimator": "PCA", - "estimator_params": { "svd_solver": "auto", "n_components": 3 } - }, + "algorithm": [ { "estimator": "KMeans", "estimator_params": { "n_init": 10, "n_clusters": "[SPECIAL_VALUE]auto" }, "estimator_methods": { "inference": "predict" } + }, + { + "estimator": "DBSCAN", + "estimator_params": { "eps": 3, "min_samples": 5 } } ] }, @@ -65,21 +69,19 @@ "estimator": ["KNeighborsClassifier", "KNeighborsRegressor"], "estimator_params": { "n_neighbors": 5, "algorithm": "brute" } }, - { "estimator": "ElasticNet" }, { "estimator": "SVC" } ] } }, "TEMPLATES": { - "multi clsf": { - "SETS": ["common", "multi clsf data"], - "algorithm": { "estimator": "LogisticRegression" } - }, "supervised": { "SETS": ["common", "binary clsf data", "supervised algorithms"] }, "unsupervised": { "SETS": ["common", "blobs data", "unsupervised algorithms"] + }, + "regression": { + "SETS": ["common", "regression data", "regression algorithms"] } } } From c99d041294309f28a0353a64c8dc58d88705569d Mon Sep 17 00:00:00 2001 From: ethanglaser Date: Fri, 24 Apr 2026 08:51:02 -0700 Subject: [PATCH 2/2] address comments --- configs/sklearn_example.json | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/sklearn_example.json b/configs/sklearn_example.json index 76c7422c..b840b9e6 100644 --- a/configs/sklearn_example.json +++ b/configs/sklearn_example.json @@ -67,7 +67,7 @@ "algorithm": [ { "estimator": ["KNeighborsClassifier", "KNeighborsRegressor"], - "estimator_params": { "n_neighbors": 5, "algorithm": "brute" } + "estimator_params": { "n_neighbors": 5, "algorithm": "kd_tree" } }, { "estimator": "SVC" } ]