scikit-learn_bench/configs/spmd/large_scale/spmd_for_online.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
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
{
    "INCLUDE": ["../../common/sklearn.json", "large_scale.json"],
    "PARAMETERS_SETS": {
        "covariance data": {
            "data": [
                {
                    "source": "make_blobs",
                    "generation_kwargs": {
                        "centers": 1,
                        "n_samples": 1000000,
                        "n_features": [10, 100]
                    },
                    "split_kwargs": {"ignore": true}
                }
            ]
        },
        "basic_statistics data": {
            "data": {
                "source": "make_blobs",
                "generation_kwargs": {
                    "centers": 1,
                    "n_samples": 1000000,
                    "n_features": [10, 100]
                },
                "split_kwargs": {"ignore": true}
            }
        },
        "linear_regression data": {
            "data": {
                "source": "make_regression",
                "split_kwargs": {"train_size": 0.2, "test_size": 0.8},
                "generation_kwargs": {
                    "n_samples": 1000000,
                    "n_features": [10, 100],
                    "n_informative": 5,
                    "noise": 2.0
                }
            }
        },
        "pca data": {
            "data": {
                "source": "make_blobs",
                "generation_kwargs": {
                    "centers": 1,
                    "n_samples": 1000000,
                    "n_features": [10, 100]
                },
                "split_kwargs": {"ignore": true}
            }
        },
        "basic_statistics": {
            "algorithm": [
                {
                    "estimator": "BasicStatistics",
                    "library": "sklearnex.spmd",
                    "estimator_methods": {"training": "fit"}
                }
            ]
        },
        "covariance": {
            "algorithm": [
                {
                    "estimator": "EmpiricalCovariance",
                    "library": "sklearnex.spmd",
                    "estimator_methods": {"training": "fit"}
                }
            ]
        },
        "linear_regression": {
            "algorithm": [
                {
                    "estimator": "LinearRegression",
                    "library": "sklearnex.spmd",
                    "estimator_methods": {"training": "fit"}
                }
            ]
        },
        "pca": {
            "algorithm": [
                {
                    "estimator": "PCA",
                    "library": "sklearnex.spmd",
                    "estimator_methods": {"training": "fit", "inference": ""}
                }
            ]
        }
    },
    "TEMPLATES": {
        "basic_statistics": {"SETS": ["basic_statistics", "basic_statistics data", "sklearnex spmd implementation", "large scale full one node parameters"]},
        "covariance": {"SETS": ["covariance", "covariance data", "sklearnex spmd implementation", "large scale full one node parameters"]},
        "linear_regression": {
            "SETS": ["linear_regression", "linear_regression data", "sklearnex spmd implementation", "large scale full one node parameters"]
        },
        "pca": {"SETS": ["pca", "pca data", "sklearnex spmd implementation", "large scale full one node parameters"]}
    }
}