Skip to content

Commit 8897c6d

Browse files
committed
use 100 instead of 800 n_estimators; looks the same but faster; ESLII uses 800
1 parent a35bfd5 commit 8897c6d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/ensemble/plot_partial_dependence.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@
6767

6868
print('_' * 80)
6969
print("Training GBRT...")
70-
clf = GradientBoostingRegressor(n_estimators=800, max_depth=4,
70+
clf = GradientBoostingRegressor(n_estimators=100, max_depth=4,
7171
learning_rate=0.1, loss='huber',
7272
random_state=1)
7373
clf.fit(X_train, y_train)

0 commit comments

Comments
 (0)