|
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