scikit-learn_bench/configs/regular/bf16/linear_model.json at 2edb59726027019615928c71a44929ba5d3902f7 · 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
{
    "INCLUDE": ["../../common/sklearn.json"],
    "PARAMETERS_SETS": {
        "synthetic data": {
            "data": [
                { "source": "make_regression", "generation_kwargs": { "n_samples": 3005000,  "n_features": 10, "noise": 1.25 }, "split_kwargs": { "train_size": 3000000, "test_size": 5000 } }
            ]
        },
        "common linear parameters": {
            "algorithm": {
                "estimator": "LinearRegression",
                "estimator_params": { "fit_intercept": true, "copy_X": true }
            },
            "data": {
                "dtype": ["float32"],
                "order": "C"
            }
        },
        "sklearn linear parameters": {
            "estimator_params": { "n_jobs": "[SPECIAL_VALUE]physical_cpus" }
        }
    },
    "TEMPLATES": {
        "sklearn linear": {
            "SETS": [
                "sklearn-ex[gpu] implementations",
                "common linear parameters",
                "sklearn linear parameters",
                "synthetic data"
            ]
        }
    }
}