Skip to content

Commit d40389e

Browse files
forest regression
1 parent 902f0ec commit d40389e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

configs/spmd/large_scale/forest_reg.json

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,9 @@
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
},

0 commit comments

Comments
 (0)