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

"featue_map" typo changed to "feature_map" (#6540)

Browse Source
This commit is contained in:
MBSMachineLearning 2020-12-21 17:11:11 +03:00 committed by GitHub
parent fbb980d9d3
commit 95cbfad990
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
demo/guide-python/feature_weights.py
View File

@ -28,10 +28,10 @@ def main(args):
'colsample_bynode': 0.5}, 'colsample_bynode': 0.5},
dtrain, num_boost_round=10, dtrain, num_boost_round=10,
evals=[(dtrain, 'd')]) evals=[(dtrain, 'd')])
featue_map = bst.get_fscore() feature_map = bst.get_fscore()
# feature zero has 0 weight # feature zero has 0 weight
assert featue_map.get('f0', None) is None assert feature_map.get('f0', None) is None
assert max(featue_map.values()) == featue_map.get('f9') assert max(feature_map.values()) == feature_map.get('f9')
if args.plot: if args.plot:
xgboost.plot_importance(bst) xgboost.plot_importance(bst)