scikit-learn_bench/configs/spmd/large_scale/incremental/linear_model.json at 72d65c17e8495bce04c335bf0dc7426f4d92ed5e · 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
{
    "INCLUDE": ["../../../common/sklearn.json", "../../../regular/linear_model.json", "../large_scale.json"],
    "PARAMETERS_SETS": {
        "spmd linear parameters": {
            "algorithm": {
                "estimator": "IncrementalLinearRegression",
                "estimator_methods": { "training": "fit" },
                "num_batches": {"training": 10}
            }
        },
        "synthetic data": {
            "data": [
		        { "source": "make_regression", "generation_kwargs": { "n_samples": 305000,  "n_features": 1000, "noise": 1.25 }, "split_kwargs": { "train_size": 300000, "test_size": 5000 } }
            ]
        }
    },
    "TEMPLATES": {
        "linreg": {
            "SETS": [
                "sklearnex spmd implementation",
                "large scale 32 parameters",
		"synthetic data",
                "spmd linear parameters"
            ]
        }
    }
}