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

Convert numpy float to Python float in feat score. (#7047)

Browse Source
This commit is contained in:
Jiaming Yuan 2021-06-21 20:58:43 +08:00 committed by GitHub
parent bbfffb444d
commit da1ad798ca
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python-package/xgboost/core.py
View File

@ -2235,7 +2235,7 @@ class Booster(object):
scores_arr = ctypes2numpy(scores, length.value, np.float32) scores_arr = ctypes2numpy(scores, length.value, np.float32)
results = {} results = {}
for feat, score in zip(features_arr, scores_arr): for feat, score in zip(features_arr, scores_arr):
results[feat] = score results[feat] = float(score)
return results return results
def trees_to_dataframe(self, fmap=''): def trees_to_dataframe(self, fmap=''):

5
tests/python/test_basic.py
View File

@ -171,6 +171,11 @@ class TestBasic:
with pytest.raises(ValueError): with pytest.raises(ValueError):
booster.get_fscore() booster.get_fscore()
booster.feature_names = None
# Use JSON to make sure the output has native Python type
scores = json.loads(json.dumps(booster.get_fscore()))
np.testing.assert_allclose(scores["f0"], 6.0)
def test_load_file_invalid(self): def test_load_file_invalid(self):
with pytest.raises(xgb.core.XGBoostError): with pytest.raises(xgb.core.XGBoostError):
xgb.Booster(model_file='incorrect_path') xgb.Booster(model_file='incorrect_path')