scikit-learn_bench/configs/spmd/large_scale/kmeans_strong.json at af48e968488fbfa2c4b664e7fd0ec973ce90bd4e · 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
{
    "INCLUDE": ["../../common/sklearn.json", "../large_scale/large_scale.json"],
    "PARAMETERS_SETS": {
        "spmd kmeans parameters": {
            "algorithm": {
                "estimator": "KMeans",
                "estimator_params": {
                    "algorithm": "lloyd",
		    "max_iter": 20,
		    "n_clusters": 100
                },
                "estimator_methods": { "training": "fit", "inference": "predict" },
                "sklearnex_context": { "use_raw_input": true }
            }
        },
        "synthetic data": {
                "data": [
                        { "source": "make_blobs", "generation_kwargs": { "n_samples": 25000000,  "n_features": 100, "centers": 100 }}
                ]
        }
    },
    "TEMPLATES": {
        "kmeans": {
            "SETS": [
                "synthetic data",
                "sklearnex spmd implementation",
                "large scale strong <=64 parameters",
                "spmd kmeans parameters"
            ]
        }
    }
}