Skip to content

Commit

Permalink
rotation
Browse files Browse the repository at this point in the history
  • Loading branch information
ldirer committed Jul 20, 2014
1 parent 84af9c3 commit 2f6406e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions sklearn/ensemble/tests/test_forest.py
Original file line number Diff line number Diff line change
Expand Up @@ -679,15 +679,15 @@ def check_warm_start(name, X, y, random_state=42):
for n_estimators in [5, 10]:
if clf_ws is None:
clf_ws = ForestEstimator(n_estimators=n_estimators,
random_state=random_state, n_jobs=1,
random_state=random_state,
warm_start=True)
else:
clf_ws.set_params(n_estimators=n_estimators)
clf_ws.fit(X, y)
assert_equal(len(clf_ws), n_estimators)

clf_no_ws = ForestEstimator(n_estimators=10, random_state=random_state,
n_jobs=1, warm_start=False)
warm_start=False)
clf_no_ws.fit(X, y)
assert_array_equal(clf_ws.apply(X), clf_no_ws.apply(X),
err_msg="Failed with {0}".format(name))
Expand Down

0 comments on commit 2f6406e

Please sign in to comment.