|
1 |
| -{ |
| 1 | +{ "INCLUDE": ["./common/sklearn.json"], |
2 | 2 | "PARAMETERS_SETS": {
|
3 | 3 | "common": {"bench": {"n_runs": 10, "time_limit": 60}},
|
4 | 4 | "covariance data": {
|
|
7 | 7 | "source": "make_blobs",
|
8 | 8 | "generation_kwargs": {
|
9 | 9 | "centers": 1,
|
10 |
| - "n_samples": 1000, |
11 |
| - "n_features": [16, 64] |
| 10 | + "n_samples": 12000000, |
| 11 | + "n_features": [10, 100] |
12 | 12 | },
|
13 | 13 | "split_kwargs": {"ignore": true}
|
14 | 14 | }
|
|
19 | 19 | "source": "make_blobs",
|
20 | 20 | "generation_kwargs": {
|
21 | 21 | "centers": 1,
|
22 |
| - "n_samples": 10000, |
23 |
| - "n_features": [16, 64] |
| 22 | + "n_samples": 12000000, |
| 23 | + "n_features": [10, 100] |
24 | 24 | },
|
25 | 25 | "split_kwargs": {"ignore": true}
|
26 | 26 | }
|
|
30 | 30 | "source": "make_regression",
|
31 | 31 | "split_kwargs": {"train_size": 0.2, "test_size": 0.8},
|
32 | 32 | "generation_kwargs": {
|
33 |
| - "n_samples": 5000, |
34 |
| - "n_features": [40, 100], |
| 33 | + "n_samples": 12000000, |
| 34 | + "n_features": [10, 100], |
35 | 35 | "n_informative": 5,
|
36 | 36 | "noise": 2.0
|
37 | 37 | }
|
|
42 | 42 | "source": "make_blobs",
|
43 | 43 | "generation_kwargs": {
|
44 | 44 | "centers": 1,
|
45 |
| - "n_samples": 1000, |
46 |
| - "n_features": [16, 64] |
| 45 | + "n_samples": 12000000, |
| 46 | + "n_features": [10, 100] |
47 | 47 | },
|
48 | 48 | "split_kwargs": {"ignore": true}
|
49 | 49 | }
|
|
54 | 54 | "estimator": "IncrementalEmpiricalCovariance",
|
55 | 55 | "library": "sklearnex.covariance",
|
56 | 56 | "estimator_methods": {"training": "partial_fit"},
|
57 |
| - "num_batches": {"training": 2} |
| 57 | + "num_batches": {"training": 12} |
58 | 58 | }
|
59 | 59 | ]
|
60 |
| - }, |
| 60 | + }, |
61 | 61 | "basic_statistics": {
|
62 | 62 | "algorithm": [
|
63 | 63 | {
|
64 | 64 | "estimator": "IncrementalBasicStatistics",
|
65 | 65 | "library": "sklearnex.basic_statistics",
|
66 |
| - "num_batches": {"training": 2} |
| 66 | + "estimator_methods": {"training": "partial_fit"}, |
| 67 | + "num_batches": {"training": 12} |
67 | 68 | }
|
68 | 69 | ]
|
69 | 70 | },
|
|
72 | 73 | {
|
73 | 74 | "estimator": "IncrementalLinearRegression",
|
74 | 75 | "library": "sklearnex.linear_model",
|
75 |
| - "num_batches": {"training": 2} |
| 76 | + "estimator_methods": {"training": "partial_fit"}, |
| 77 | + "num_batches": {"training": 12} |
76 | 78 | }
|
77 | 79 | ]
|
78 | 80 | },
|
|
81 | 83 | {
|
82 | 84 | "estimator": "IncrementalPCA",
|
83 | 85 | "library": "sklearnex.preview.decomposition",
|
84 |
| - "num_batches": {"training": 2} |
85 |
| - } |
| 86 | + "estimator_methods": {"training": "partial_fit"}, |
| 87 | + "num_batches": {"training": 12} |
| 88 | + } |
86 | 89 | ]
|
87 | 90 | }
|
88 | 91 | },
|
89 | 92 | "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"]}, |
91 | 95 | "linear_regression": {
|
92 |
| - "SETS": ["common", "linear_regression", "linear_regression data"] |
| 96 | + "SETS": ["common", "linear_regression", "linear_regression data", "sklearn-ex[gpu] implementations"] |
93 | 97 | },
|
94 |
| - "pca": {"SETS": ["common", "pca", "pca data"]} |
| 98 | + "pca": {"SETS": ["common", "pca", "pca data", "sklearn-ex[gpu] implementations"]} |
95 | 99 | }
|
96 | 100 | }
|
0 commit comments