scikit-learn_bench/configs/regular/bf16/forest.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
{
    "INCLUDE": ["../../common/sklearn.json"],
    "PARAMETERS_SETS": {
        "common forest params": {
            "data": {
                "dtype": ["float32"]
            }
        },
        "forest classifier params": {
            "algorithm": {"estimator": "RandomForestClassifier"},
	    "data": { "source": "make_classification", "split_kwargs": { "train_size": 500000, "test_size": 1000 },    "generation_kwargs": {  "n_samples": 501000, "n_features": 10, "n_classes": 2 },  "algorithm": { "estimator_params": { "n_estimators": 20, "max_depth": 4 } } }
        },
	"forest regression params": {
            "algorithm": {"estimator": "RandomForestRegressor"},
            "data": { "source": "make_regression", "generation_kwargs": { "n_samples": 501000,  "n_features": 10, "noise": 1.25 }, "split_kwargs": { "train_size": 500000, "test_size": 1000 }, "algorithm": { "estimator_params": { "n_estimators": 20, "max_depth": 4 } }}
        }
    },
    "TEMPLATES": {
        "forest cls": {
            "SETS": [
                "sklearn-ex[gpu] implementations",
                "common forest params",
                "forest classifier params"
            ]
        },
	"forest reg": {
            "SETS": [
                "sklearn-ex[gpu] implementations",
                "common forest params",
                "forest regression params"
            ]
        }
    }
}