Fix model slicing. (#7149)

* Use correct pointer. * Remove best_iteration/best_score.
2021-08-03 11:51:56 +08:00
parent 36346f8f56
commit d080b5a953
6 changed files with 37 additions and 5 deletions
--- a/tests/python/test_cli.py
+++ b/tests/python/test_cli.py
@@ -99,6 +99,8 @@ eval[test] = {data_path}
            # CLI model doesn't contain feature info.
            booster.feature_names = None
            booster.feature_types = None
+            booster.set_attr(best_iteration=None)
+            booster.set_attr(best_ntree_limit=None)

            booster.save_model(model_out_py)
            py_predt = booster.predict(data)