Skip to content

Commit 902f0ec

Browse files
forest regression
1 parent 4ac119e commit 902f0ec

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

configs/spmd/large_scale/forest_strong_reg.json

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,15 +8,17 @@
88
},
99
"synthetic data": {
1010
"data": [
11-
{ "source": "make_regression", "generation_kwargs": { "n_samples": 10000000, "n_features": 100, "noise": 1.25 }, "split_kwargs": { "train_size": 10000000, "test_size": 5000 }, "algorithm": { "estimator_params": { "n_estimators": 20, "max_depth": 4 } }}
11+
{ "source": "make_regression", "generation_kwargs": { "n_samples": 10000000, "n_features": 10, "noise": 1.25 }, "split_kwargs": { "train_size": 9090000, "test_size": 100000 }, "algorithm": { "estimator_params": { "n_estimators": 5, "max_depth": 4 } }},
12+
{ "source": "make_regression", "generation_kwargs": { "n_samples": 100000, "n_features": 1000, "noise": 1.25 }, "split_kwargs": { "train_size": 90000, "test_size": 10000 }, "algorithm": { "estimator_params": { "n_estimators": 10, "max_depth": 4 } }}
13+
1214
]
1315
}
1416
},
1517
"TEMPLATES": {
1618
"forestReg": {
1719
"SETS": [
1820
"sklearnex spmd implementation",
19-
"large scale strong 2k parameters",
21+
"large scale strong 32 parameters",
2022
"synthetic data",
2123
"spmd forest regression parameters"
2224
]

0 commit comments

Comments
 (0)