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