scikit-learn_bench/configs/spmd/large_scale/dbscan_strong.json at e684c8de56be1f5f50fb33f421ebb574f46d545f · 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
{
    "INCLUDE": ["../../common/sklearn.json", "../../regular/dbscan.json", "large_scale.json"],
    "PARAMETERS_SETS": {
        "spmd dbscan parameters": {
	    "algorithm": {
                "estimator": "DBSCAN",
                "estimator_methods": {
                "training": "fit"
		},
                "estimator_params" : {
                        "eps": 15, "min_samples": 50
                },
		"sklearnex_context": { "use_raw_input": true }
	    },
	    "data": {
		"dtype": "float64"
	    }
	},
	"synthetic dataset": {
            "data": [
                { "source": "make_blobs", "generation_kwargs": { "n_samples": 4000000,  "n_features": 100, "centers": 10 } }
            ]
	}
    },
    "TEMPLATES": {
        "dbscan": {
            "SETS": [
                "common dbscan parameters",
                "synthetic dataset",
                "sklearnex spmd implementation",
                "large scale strong <=64 parameters",
                "spmd dbscan parameters"
            ]
        }
    }
}