scikit-learn_bench/configs/regular/linear_model.json at main · 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
{
    "INCLUDE": ["../common/sklearn.json", "../common/linear_model.json"],
    "PARAMETERS_SETS": {
        "regression datasets": [
            {
                "data": {
                    "source": "make_regression",
                    "split_kwargs": { "train_size": 0.2, "test_size": 0.8 },
                    "generation_kwargs": [
                        {
                            "n_samples": 5000000,
                            "n_features": 50,
                            "n_informative": 5,
                            "noise": 20.0
                        },
                        {
                            "n_samples": 500000,
                            "n_features": 400,
                            "n_informative": 5,
                            "noise": 40.0
                        },
                        {
                            "n_samples": 100000,
                            "n_features": 2000,
                            "n_informative": 5,
                            "noise": 60.0
                        }
                    ]
                }
            },
            {
                "data": {
                    "dataset": "year_prediction_msd",
                    "preprocessing_kwargs": { "normalize": "standard" },
                    "split_kwargs": { "train_size": 0.5, "test_size": 0.5 }
                }
            },
            {
                "data": {
                    "dataset": ["hepmass", "susy"],
                    "split_kwargs": { "train_size": 2000000, "test_size": null }
                }
            }
        ]
    },
    "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"]
        }
    }
}