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

Fix for Feature shape mismatch error (#7715)

Browse Source
This commit is contained in:
Pradipta Ghosh 2022-03-03 19:06:29 +05:30 committed by GitHub
parent a92e0f6240
commit 68b6d6bbe2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 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

@ -2242,7 +2242,7 @@ class Booster:
def num_features(self) -> int: def num_features(self) -> int:
'''Number of features in booster.''' '''Number of features in booster.'''
features = ctypes.c_int() features = c_bst_ulong()
assert self.handle is not None assert self.handle is not None
_check_call(_LIB.XGBoosterGetNumFeature(self.handle, ctypes.byref(features))) _check_call(_LIB.XGBoosterGetNumFeature(self.handle, ctypes.byref(features)))
return features.value return features.value