1+ {
2+ "INCLUDE" : [" ./common/sklearn.json" ],
3+ "PARAMETERS_SETS" : {
4+ "common" : {"bench" : {"n_runs" : 10 , "time_limit" : 60 }},
5+ "covariance data" : {
6+ "data" : [
7+ {
8+ "source" : " make_blobs" ,
9+ "generation_kwargs" : {
10+ "centers" : 1 ,
11+ "n_samples" : 1200000 ,
12+ "n_features" : [10 , 100 ]
13+ },
14+ "split_kwargs" : {"ignore" : true }
15+ }
16+ ]
17+ },
18+ "basic_statistics data" : {
19+ "data" : {
20+ "source" : " make_blobs" ,
21+ "generation_kwargs" : {
22+ "centers" : 1 ,
23+ "n_samples" : 1200000 ,
24+ "n_features" : [10 , 100 ]
25+ },
26+ "split_kwargs" : {"ignore" : true }
27+ }
28+ },
29+ "linear_regression data" : {
30+ "data" : {
31+ "source" : " make_regression" ,
32+ "split_kwargs" : {"train_size" : 0.2 , "test_size" : 0.8 },
33+ "generation_kwargs" : {
34+ "n_samples" : 1200000 ,
35+ "n_features" : [10 , 100 ],
36+ "n_informative" : 5 ,
37+ "noise" : 2.0
38+ }
39+ }
40+ },
41+ "pca data" : {
42+ "data" : {
43+ "source" : " make_blobs" ,
44+ "generation_kwargs" : {
45+ "centers" : 1 ,
46+ "n_samples" : 1200000 ,
47+ "n_features" : [10 , 100 ]
48+ },
49+ "split_kwargs" : {"ignore" : true }
50+ }
51+ },
52+ "covariance" : {
53+ "algorithm" : [
54+ {
55+ "estimator" : " EmpiricalCovariance" ,
56+ "library" : " sklearnex.covariance" ,
57+ "estimator_methods" : {"training" : " fit" }
58+ }
59+ ]
60+ },
61+ "basic_statistics" : {
62+ "algorithm" : [
63+ {
64+ "estimator" : " BasicStatistics" ,
65+ "library" : " sklearnex.basic_statistics" ,
66+ "estimator_methods" : {"training" : " fit" }
67+ }
68+ ]
69+ },
70+ "linear_regression" : {
71+ "algorithm" : [
72+ {
73+ "estimator" : " LinearRegression" ,
74+ "library" : " sklearnex.linear_model" ,
75+ "estimator_methods" : {"training" : " fit" }
76+ }
77+ ]
78+ },
79+ "pca" : {
80+ "algorithm" : [
81+ {
82+ "estimator" : " IncrementalPCA" ,
83+ "library" : " sklearnex.decomposition" ,
84+ "estimator_methods" : {"training" : " fit" }
85+ }
86+ ]
87+ }
88+ },
89+ "TEMPLATES" : {
90+ "basic_statistics" : {"SETS" : [" common" , " basic_statistics" , " basic_statistics data" , " sklearn-ex[gpu] implementations" ]},
91+ "covariance" : {"SETS" : [" common" , " covariance" , " covariance data" , " sklearn-ex[gpu] implementations" ]},
92+ "linear_regression" : {
93+ "SETS" : [" common" , " linear_regression" , " linear_regression data" , " sklearn-ex[gpu] implementations" ]
94+ },
95+ "pca" : {"SETS" : [" common" , " pca" , " pca data" , " sklearn-ex[gpu] implementations" ]}
96+ }
97+ }
0 commit comments