sklearn multiclass svm 函数

我有多个类别的标签,想计算我模型的准确率。
我对需要使用哪个 sklearn 函数感到有些困惑。据我所知,下面的代码仅用于二元分类。

# 将X, y分为训练和测试数据
X_train, X_test, y_train, y_test = train_test_split(X, y,  test_size=0.25,random_state = 0)
# 训练一个线性SVM分类器
from sklearn.svm import SVC
svm_model_linear = SVC(kernel = 'linear', C = 1).fit(X_train, y_train)
svm_predictions = svm_model_linear.predict(X_test)
# 计算X_test的模型准确率
accuracy = svm_model_linear.score(X_test, y_test)
print accuracy

根据以下链接的理解:使用 OneVsRestClassifier 时,sklearn.svm.SVC 的 decision_function_shape 应选择哪种?

对于多类别分类,我应该使用 OneVsRestClassifier 并设置 decision_function_shape(使用 ovrovo,并检查哪种效果更好)

svm_model_linear = OneVsRestClassifier(SVC(kernel = 'linear',C = 1, decision_function_shape = 'ovr')).fit(X_train, y_train)

主要问题是预测标签的时间对我很重要,但运行分类器并预测数据需要大约1分钟(此时间还包括像PCA这样的特征降维的时间)?有什么建议可以减少SVM多分类器的时间吗?


回答:

这里有几点需要考虑:

1) 你看,OneVsRestClassifier 会将所有标签分开,并在给定数据上训练多个svm对象(每个标签一个)。所以每次,只会向单个svm对象提供二元数据。

2) SVC 内部使用 libsvmliblinear,它们对于多类别或多标签输出采用 ‘OvO’ 策略。但由于第1点,这个点没有用。因为 libsvm 只会得到二元数据。

即使它确实如此,它也不考虑 'decision_function_shape'。因此,无论你提供 decision_function_shape = 'ovr' 还是 decision_function_shape = 'ovr' 都没有关系。

所以看起来你对问题理解有误。decision_function_shape 不应该影响速度。尝试在拟合前标准化你的数据。SVM 对标准化数据表现良好。

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中创建了一个多类分类项目。该项目可以对…

发表回复

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