Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from sklearn.feature_selection import SelectFromModel
- from sklearn.ensemble import GradientBoostingClassifier
- #GBDT as the feature selection of the base model
- SelectFromModel(GradientBoostingClassifier()).fit_transform(iris.data, iris.target)
Add Comment
Please, Sign In to add comment