Don't set_params at the end of set_state. (#4947)

* Don't set_params at the end of set_state.

* Also fix another issue found in dask prediction.

* Add note about prediction.

Don't support other prediction modes at the moment.
This commit is contained in:
Jiaming Yuan
2019-10-15 10:08:26 -04:00
committed by GitHub
parent 2ebdec8aa6
commit 7e72a12871
6 changed files with 70 additions and 8 deletions

View File

@@ -32,6 +32,7 @@ def main(client):
# you can pass output directly into `predict` too.
prediction = xgb.dask.predict(client, bst, dtrain)
prediction = prediction.compute()
print('Evaluation history:', history)
return prediction