scikit-learn_bench/configs/regular/logreg_cv.json at 7839432265b0ca6d83a0b9e584245d2f967f1f89 · 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
{
    "INCLUDE": ["../common/sklearn.json"],
    "PARAMETERS_SETS": {
        "sklearn logreg_cv parameters": {
            "algorithm": {
                "estimator": "LogisticRegressionCV",
                "library": ["sklearn", "sklearnex"],
                "device": "cpu",
                "estimator_methods": { "inference": "predict" },
                "estimator_params": {
                    "Cs": 10,
                    "solver": "lbfgs",
                    "max_iter": 1000,
                    "random_state": 123
                }
            }
        },
        "logreg_cv datasets": [
            {
                "data": {
                    "source": "make_classification",
                    "generation_kwargs": [
                        {
                            "n_samples": 100000,
                            "n_features": 1000,
                            "n_informative": 800,
                            "n_classes": [2]
                        }
                    ],
                    "split_kwargs": {
                        "train_size": 0.05,
                        "test_size": 0.95
                    }
                }
            }
        ]
    },
    "TEMPLATES": {
        "sklearn logreg_cv": {
            "SETS": [
                "sklearn-ex[cpu,gpu] implementations",
                "sklearn logreg_cv parameters",
                "logreg_cv datasets"
            ]
        }
    }
}