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

2021-06-21 20:58:43 +08:00
parent bbfffb444d
commit da1ad798ca
2 changed files with 6 additions and 1 deletions
--- a/python-package/xgboost/core.py
+++ b/python-package/xgboost/core.py
@@ -2235,7 +2235,7 @@ class Booster(object):
        scores_arr = ctypes2numpy(scores, length.value, np.float32)
        results = {}
        for feat, score in zip(features_arr, scores_arr):
-            results[feat] = score
+            results[feat] = float(score)
        return results

    def trees_to_dataframe(self, fmap=''):