scikit-learn_bench/configs/weekly/linear_model.json at 098b21b041c65308ab5017f284c8dead4dc07413 · IntelPython/scikit-learn_bench · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
{
    "INCLUDE": ["../common/sklearn.json", "../common/linear_model.json"],
    "PARAMETERS_SETS": {
        "regression datasets": [
            {
                "data": {
                    "source": "make_regression",
                    "split_kwargs": { "train_size": 0.5, "test_size": 0.5 },
                    "generation_kwargs": [
                        {
                            "n_samples": 20000000, "n_features": 50,
                            "n_informative": 5, "noise": 20.0
                        },
                        {
                            "n_samples": 2000000, "n_features": 400,
                            "n_informative": 5, "noise": 40.0
                        },
                        {
                            "n_samples": 200000, "n_features": 5000,
                            "n_informative": 5, "noise": 80.0
                        }
                    ]
                }
            },
            {
                "data": {
                    "dataset": [
                        "yolanda",
                        "hepmass",
                        "susy"
                    ],
                    "preprocessing_kwargs": {
                        "normalize": true
                    },
                    "split_kwargs": { "ignore": true }
                }
            }
        ]
    },
    "TEMPLATES": {
        "sklearn linear": {
            "SETS": [
                "sklearn-ex[cpu,gpu] implementations",
                "common linear parameters",
                "sklearn linear parameters",
                "regression datasets"
            ]
        },
        "sklearn ridge": {
            "SETS": [
                "sklearn-ex[cpu] implementations",
                "common ridge parameters",
                "sklearn ridge parameters",
                "regression datasets"
            ]
        },
        "sklearn lasso": {
            "SETS": [
                "sklearn-ex[cpu] implementations",
                "common lasso parameters",
                "regression datasets"
            ]
        },
        "sklearn elasticnet": {
            "SETS": [
                "sklearn-ex[cpu] implementations",
                "common elasticnet parameters",
                "regression datasets"
            ]
        },
        "cuml linear": {
            "SETS": [
                "cuml implementation",
                "common linear parameters",
                "cuml L2 parameters",
                "regression datasets"
            ]
        },
        "cuml ridge": {
            "SETS": [
                "cuml implementation",
                "common ridge parameters",
                "cuml L2 parameters",
                "regression datasets"
            ]
        },
        "cuml lasso": {
            "SETS": [
                "cuml implementation",
                "common lasso parameters",
                "cuml L1 parameters",
                "regression datasets"
            ]
        },
        "cuml elasticnet": {
            "SETS": [
                "cuml implementation",
                "common elasticnet parameters",
                "cuml L1 parameters",
                "regression datasets"
            ]
        }
    }
}