mlxtend.feature_selection 前向选择在使用SVM线性核时不起作用?

我在使用mlxtend包的SVM进行特征选择。X是一个包含特征的数据框,y是目标变量。这是我的代码的一部分。

from sklearn.svm import SVCfrom mlxtend.feature_selection import SequentialFeatureSelector as SFSdef SFFS(X, y, C_GS, gamma_GS, kernel_GS):    sfs = SFS(SVC(kernel = kernel_GS, C = C_GS, gamma = gamma_GS),         k_features = (1, num_of_features),          forward= True,          floating = False,          verbose= 2,          scoring= 'roc_auc',          #scoring= 'accuracy',          cv = 10,          n_jobs= -1         ).fit(X, y)    return sfsdef SFFS_lin(X, y, C_GS, kernel_GS):    sfs = SFS(SVC(kernel = kernel_GS, C = C_GS),         k_features = (1, num_of_features),          forward= True,          floating = False,          verbose= 2,          scoring= 'roc_auc',          cv = 10,          n_jobs= -1         ).fit(X, y)    return sfsdef featureNames(sfs):    Feature_Names = sfs.k_feature_names_    return Feature_Namessfs_lin = SFFS_lin(X, y, 1,'linear')#sfs_rbf = SFFS(X, y, 1, 'auto', 'rbf')names = featureNames(sfs_lin)print(names)

代码开始运行,但不久后就在这里卡住了:

[Parallel(n_jobs=-1)]: Using backend LokyBackend with 8 concurrentworkers. [Parallel(n_jobs=-1)]: Done 28 out of 28 | elapsed: 2.5sremaining: 0.0s [Parallel(n_jobs=-1)]: Done 28 out of 28 |elapsed: 2.5s finished

[2021-01-24 00:01:57] Features: 1/28 — score:0.6146428161908037[Parallel(n_jobs=-1)]: Using backend LokyBackend with 8 concurrent workers.

当使用rbf核时,代码运行得很顺畅。如果我将函数改为通过设置forward参数为False来执行向后消除,

forward=False,

它也运行得很顺畅。卡住的问题似乎出现在使用线性核进行前向选择时。这是愚蠢的错误还是我错过了什么琐碎的东西?

系统信息:

Python 3.8.5scikit-learn 0.24.1mlxtend 0.18.0

回答:

看来这只是一个愚蠢的错误。

将交叉验证参数

cv = 10

改为9后,它就运行了..

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注