scikit-learn_bench/configs/regular/kmeans.json at add56d869d97cd41b42dd0a7fc923949292b94f3 · 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
{
    "INCLUDE": ["../common/sklearn.json"],
    "PARAMETERS_SETS": {
        "common kmeans parameters": {
            "algorithm": {
                "estimator": "KMeans",
                "estimator_params": {
                    "n_clusters": "[SPECIAL_VALUE]auto",
                    "n_init": 1,
                    "max_iter": 30,
                    "tol": 1e-3,
                    "random_state": 42
                },
                "estimator_methods": { "inference": "predict" }
            },
            "data": {
                "dtype": ["float32", "float64"],
                "preprocessing_kwargs": { "normalize": true }
            }
        },
        "sklearn kmeans parameters": {
            "algorithm": { "estimator_params": { "init": "k-means++", "algorithm": "lloyd" } }
        },
        "cuml kmeans parameters": {
            "algorithm": {
                "estimator_params": { "init": "scalable-k-means++" }
            }
        },
        "kmeans datasets": [
            {
                "data": [
                    {
                        "dataset": ["covtype", "sensit"],
                        "split_kwargs": { "ignore": true }
                    },
                    {
                        "dataset": ["mnist", "gisette"],
                        "split_kwargs": { "ignore": true },
                        "preprocessing_kwargs": { "normalize": false }
                    }
                ]
            },
            {
                "data": {
                    "dataset": "higgs",
                    "split_kwargs": {
                        "train_size": 100000,
                        "test_size": 2000000,
                        "shuffle": true,
                        "random_state": 42
                    }
                },
                "algorithm": [
                    {
                        "estimator_params": {
                            "n_clusters": 100,
                            "max_iter": 10
                        }
                    },
                    {
                        "estimator_params": {
                            "n_clusters": 10,
                            "max_iter": 100
                        }
                    }
                ]
            }
        ]
    },
    "TEMPLATES": {
        "sklearn kmeans": {
            "SETS": [
                "sklearn-ex[cpu,gpu] implementations",
                "common kmeans parameters",
                "sklearn kmeans parameters",
                "kmeans datasets"
            ]
        },
        "cuml kmeans": {
            "SETS": [
                "cuml implementation",
                "common kmeans parameters",
                "cuml kmeans parameters",
                "kmeans datasets"
            ]
        }
    }
}