scikit-learn_bench/configs/sklearnex_incremental_example.json at 5a9be80616e5dca5e50bd27145ce11c6316b4c2d · 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
{
    "PARAMETERS_SETS": {
        "common": {"bench": {"n_runs": 10, "time_limit": 60}},
        "unlabeled dataset": {
            "data": [
                {
                    "source": "make_blobs",
                    "generation_kwargs": {
                        "centers": 1,
                        "n_samples": 1000,
                        "n_features": [16, 64]
                    },
                    "split_kwargs": {"ignore": true}
                }
            ]
        },
        "labeled dataset": {
            "data": {
                "source": "make_regression",
                "split_kwargs": {"train_size": 0.2, "test_size": 0.8},
                "generation_kwargs": {
                    "n_samples": 5000,
                    "n_features": [40, 100],
                    "n_informative": 5,
                    "noise": 2.0
                }
            }
        },
        "covariance": {
            "algorithm": [
                {
                    "estimator": "IncrementalEmpiricalCovariance",
                    "library": "sklearnex.covariance",
                    "estimator_methods": {"training": "partial_fit"},
                    "num_batches": {"training": 2}
                }
            ]
        },
        "linear_regression": {
            "algorithm": [
                {
                    "estimator": "IncrementalLinearRegression",
                    "library": "sklearnex",
                    "num_batches": {"training": 2}
                }
            ]
        },
        "pca": {
            "algorithm": [
                {
                    "estimator": "IncrementalPCA",
                    "library": "sklearnex.preview",
                    "num_batches": {"training": 2}
                }
            ]
        }
    },
    "TEMPLATES": {
        "covariance": {"SETS": ["common", "covariance", "unlabeled dataset"]},
        "linear_regression": {
            "SETS": ["common", "linear_regression", "labeled dataset"]
        },
        "pca": {"SETS": ["common", "pca", "unlabeled dataset"]}
    }
}