ValueError：数组包含NaN或无穷大，同时使用scikit学习带有交叉val的递归特征消除

Question

我正在使用scikit模块递归特征消除与交叉val（RFECV）进行特征选择。 代码段如下：

svc = SVC(kernel="linear")

rfecv = RFECV(estimator=svc, step=20, cv=StratifiedKFold(y, 2),scoring='roc_auc')

rfecv.fit(X,y)

当我在sklearn / utils / validation.py中运行我的代码时出现错误ValueError: Array contains NaN or infinity。

以下检查validation.py

X.dtype.char in np.typecodes['AllFloat'] and not np.isfinite(X.sum()) and not np.isfinite(X).all() 

当我在X上测试它时，

返回False .X不是稀疏的。所以理想情况下我不应该得到这个错误。

请有人告诉我可能是错误的原因。

谢谢！