scikit-learn_bench/configs/regular/bf16/knn.json at 2edb59726027019615928c71a44929ba5d3902f7 · 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
{
    "INCLUDE": ["../../common/sklearn.json"],
    "PARAMETERS_SETS": {
        "common knn parameters": {
            "algorithm": {
                "estimator_params": {
                    "n_neighbors": [10, 100],
                    "weights": "uniform"
                }
            },
            "data": {
                "dtype": ["float32"],
                "preprocessing_kwargs": { "normalize": true }
            }
        },
        "sklearn knn parameters": {
            "algorithm": { "estimator_params": { "n_jobs": "[SPECIAL_VALUE]physical_cpus" } }
        },
        "synthetic classification data": {
	    "algorithm": {
                "estimator": "KNeighborsClassifier",
                "estimator_params": { "algorithm": "brute", "metric": "minkowski", "p": [1, 2] }
            },
            "data": [
		        { "source": "make_classification", "split_kwargs": { "train_size": 5000000, "test_size": 1000 },   "generation_kwargs": {  "n_samples": 5001000,  "n_features": 100, "n_classes": 2,  "n_informative": "[SPECIAL_VALUE]0.5" } }
            ]
        },
        "synthetic regression data": {
	    "algorithm": {
                "estimator": "KNeighborsRegressor",
                "estimator_params": { "algorithm": "brute", "metric": "minkowski", "p": [1, 2] }
            },
            "data": [
		        { "source": "make_regression", "split_kwargs": { "train_size": 5000000, "test_size": 1000 },   "generation_kwargs": {  "n_samples": 5001000,  "n_features": 100, "noise":1.5 } }
            ]
        }
    },
    "TEMPLATES": {
        "sklearn brute knn clsf": {
            "SETS": [
                "sklearn-ex[gpu] implementations",
                "common knn parameters",
                "sklearn knn parameters",
                "synthetic classification data"
            ]
        },
	"sklearn brute knn reg": {
            "SETS": [
                "sklearn-ex[gpu] implementations",
                "common knn parameters",
                "sklearn knn parameters",
                "synthetic regression data"
            ]
        }
    }
}