Skip to content

Commit c852279

Browse files
forest bf16
1 parent 4f65e1f commit c852279

File tree

2 files changed

+34
-48
lines changed

2 files changed

+34
-48
lines changed

configs/regular/bf16/ensemble.json

Lines changed: 0 additions & 48 deletions
This file was deleted.

configs/regular/bf16/forest.json

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
{
2+
"INCLUDE": ["../../common/sklearn.json"],
3+
"PARAMETERS_SETS": {
4+
"common forest params": {
5+
"data": {
6+
"dtype": ["float32"]
7+
}
8+
},
9+
"forest classifier params": {
10+
"algorithm": {"estimator": "RandomForestClassifier"},
11+
"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 } } }
12+
},
13+
"forest regression params": {
14+
"algorithm": {"estimator": "RandomForestRegressor"},
15+
"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 } }}
16+
}
17+
},
18+
"TEMPLATES": {
19+
"forest cls": {
20+
"SETS": [
21+
"sklearn-ex[gpu] implementations",
22+
"common forest params",
23+
"forest classifier params"
24+
]
25+
},
26+
"forest reg": {
27+
"SETS": [
28+
"sklearn-ex[gpu] implementations",
29+
"common forest params",
30+
"forest regression params"
31+
]
32+
}
33+
}
34+
}

0 commit comments

Comments
 (0)