将sklearn RFE与另一个软件包的估算器结合使用

时间:2018-08-03 19:41:31

标签: scikit-learn statsmodels rfe

是否可以将sklearn递归特征消除(RFE)与另一个包中的估算器一起使用?

具体来说,我想使用statsmodels包中的GLM并将其包装在sklearn RFE中吗?

如果可以,请举一些例子吗?

1 个答案:

答案 0 :(得分:2)

是的,有可能。您只需要创建一个继承 (setq exec-path-from-shell-variables '("PATH")) (exec-path-from-shell-initialize) 的类,确保它具有sklearn.base.BaseEstimatorfit方法,并确保其predict方法通过{{1 }}或fit属性。这是一个简化的类示例:

coef_