Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add base margin to sklearn interface. (#5151)

Browse Source
This commit is contained in:
Jiaming Yuan
2019-12-24 09:43:41 +08:00
committed by GitHub
parent 1d0ca49761
commit 0202e04a8e
5 changed files with 103 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/python/test_with_sklearn.py
View File

@@ -695,3 +695,23 @@ def test_XGBClassifier_resume():
assert np.any(pred1 != pred2)
assert log_loss1 > log_loss2
def test_boost_from_prediction():
from sklearn.datasets import load_breast_cancer
X, y = load_breast_cancer(return_X_y=True)
model_0 = xgb.XGBClassifier(
learning_rate=0.3, random_state=0, n_estimators=4)
model_0.fit(X=X, y=y)
margin = model_0.predict(X, output_margin=True)
model_1 = xgb.XGBClassifier(
learning_rate=0.3, random_state=0, n_estimators=4)
model_1.fit(X=X, y=y, base_margin=margin)
predictions_1 = model_1.predict(X, base_margin=margin)
cls_2 = xgb.XGBClassifier(
learning_rate=0.3, random_state=0, n_estimators=8)
cls_2.fit(X=X, y=y)
predictions_2 = cls_2.predict(X, base_margin=margin)
assert np.all(predictions_1 == predictions_2)