Skip to content

Commit 698d884

Browse files
incremental
1 parent c852279 commit 698d884

File tree

1 file changed

+22
-18
lines changed

1 file changed

+22
-18
lines changed

configs/incremental.json

Lines changed: 22 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
{
1+
{ "INCLUDE": ["./common/sklearn.json"],
22
"PARAMETERS_SETS": {
33
"common": {"bench": {"n_runs": 10, "time_limit": 60}},
44
"covariance data": {
@@ -7,8 +7,8 @@
77
"source": "make_blobs",
88
"generation_kwargs": {
99
"centers": 1,
10-
"n_samples": 1000,
11-
"n_features": [16, 64]
10+
"n_samples": 12000000,
11+
"n_features": [10, 100]
1212
},
1313
"split_kwargs": {"ignore": true}
1414
}
@@ -19,8 +19,8 @@
1919
"source": "make_blobs",
2020
"generation_kwargs": {
2121
"centers": 1,
22-
"n_samples": 10000,
23-
"n_features": [16, 64]
22+
"n_samples": 12000000,
23+
"n_features": [10, 100]
2424
},
2525
"split_kwargs": {"ignore": true}
2626
}
@@ -30,8 +30,8 @@
3030
"source": "make_regression",
3131
"split_kwargs": {"train_size": 0.2, "test_size": 0.8},
3232
"generation_kwargs": {
33-
"n_samples": 5000,
34-
"n_features": [40, 100],
33+
"n_samples": 12000000,
34+
"n_features": [10, 100],
3535
"n_informative": 5,
3636
"noise": 2.0
3737
}
@@ -42,8 +42,8 @@
4242
"source": "make_blobs",
4343
"generation_kwargs": {
4444
"centers": 1,
45-
"n_samples": 1000,
46-
"n_features": [16, 64]
45+
"n_samples": 12000000,
46+
"n_features": [10, 100]
4747
},
4848
"split_kwargs": {"ignore": true}
4949
}
@@ -54,16 +54,17 @@
5454
"estimator": "IncrementalEmpiricalCovariance",
5555
"library": "sklearnex.covariance",
5656
"estimator_methods": {"training": "partial_fit"},
57-
"num_batches": {"training": 2}
57+
"num_batches": {"training": 12}
5858
}
5959
]
60-
},
60+
},
6161
"basic_statistics": {
6262
"algorithm": [
6363
{
6464
"estimator": "IncrementalBasicStatistics",
6565
"library": "sklearnex.basic_statistics",
66-
"num_batches": {"training": 2}
66+
"estimator_methods": {"training": "partial_fit"},
67+
"num_batches": {"training": 12}
6768
}
6869
]
6970
},
@@ -72,7 +73,8 @@
7273
{
7374
"estimator": "IncrementalLinearRegression",
7475
"library": "sklearnex.linear_model",
75-
"num_batches": {"training": 2}
76+
"estimator_methods": {"training": "partial_fit"},
77+
"num_batches": {"training": 12}
7678
}
7779
]
7880
},
@@ -81,16 +83,18 @@
8183
{
8284
"estimator": "IncrementalPCA",
8385
"library": "sklearnex.preview.decomposition",
84-
"num_batches": {"training": 2}
85-
}
86+
"estimator_methods": {"training": "partial_fit"},
87+
"num_batches": {"training": 12}
88+
}
8689
]
8790
}
8891
},
8992
"TEMPLATES": {
90-
"covariance": {"SETS": ["common", "covariance", "covariance data"]},
93+
"basic_statistics": {"SETS": ["common", "basic_statistics", "basic_statistics data", "sklearn-ex[gpu] implementations"]},
94+
"covariance": {"SETS": ["common", "covariance", "covariance data", "sklearn-ex[gpu] implementations"]},
9195
"linear_regression": {
92-
"SETS": ["common", "linear_regression", "linear_regression data"]
96+
"SETS": ["common", "linear_regression", "linear_regression data", "sklearn-ex[gpu] implementations"]
9397
},
94-
"pca": {"SETS": ["common", "pca", "pca data"]}
98+
"pca": {"SETS": ["common", "pca", "pca data", "sklearn-ex[gpu] implementations"]}
9599
}
96100
}

0 commit comments

Comments
 (0)