scikit-learn_bench/configs/regular/bf16/pca.json at 9611fbc6e84e500eee3c7714a3fb665dd11cf697 · 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"],
    "PARAMETERS_SETS": {
        "pca parameters": {
            "algorithm": {
                "estimator": "PCA",
                "estimator_params": {
                    "n_components": 3,
                    "copy": true,
                    "whiten": false,
                    "svd_solver": "covariance_eigh",
                    "tol": 0.0,
                    "iterated_power": 15,
                    "random_state": 42
                }
            },
            "data": {
                "dtype": ["float32"]
            }
        },
        "synthetic data": {
            "data": [
                { "source": "make_blobs", "generation_kwargs": { "n_samples": 3000000,  "n_features": 10, "centers": 1 } }
            ]
        }
    },
    "TEMPLATES": {
        "sklearn pca": {
            "SETS": [
                "sklearn-ex[gpu] implementations",
                "pca parameters",
                "synthetic data"
            ]
        }
    }
}