Python向量机分类和回归模型应用

liftword3周前 (05-26)技术文章9

Python中的向量机模型可以应用于分类和回归问题。以下是如何在Python中使用SVM模型进行分类的示例:

from sklearn import datasets

from sklearn.model_selection import train_test_split

from sklearn import svm


# 加载鸢尾花数据集

iris = datasets.load_iris()

X = iris.data

y = iris.target


# 划分训练集和测试集

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)


# 创建并训练SVM分类器

clf = svm.SVC(kernel='linear')

clf.fit(X_train, y_train)


# 预测测试数据集的标签

y_pred = clf.predict(X_test)


# 输出性能指标

from sklearn.metrics import accuracy_score

print("Accuracy score: ", accuracy_score(y_test, y_pred))

上述示例中,我们使用了sklearn自带的鸢尾花数据集,并使用80%的数据进行训练集和20%的数据进行测试集,创建了一个线性SVM分类器,并使用训练集训练了该分类器。然后我们使用测试集来预测数据点的标签,并输出了精度分数。


如果要应用SVM模型进行回归问题,可以使用sklearn的SVR类。以下是关于如何使用SVM进行回归问题的示例:

python

from sklearn import datasets

from sklearn.model_selection import train_test_split

from sklearn import svm


# 加载糖尿病数据集

diabetes = datasets.load_diabetes()

X = diabetes.data

y = diabetes.target


# 划分训练集和测试集

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)


# 创建并训练SVM回归器

reg = svm.SVR(kernel='linear')

reg.fit(X_train, y_train)


# 预测测试数据集的标签

y_pred = reg.predict(X_test)


# 输出性能指标

from sklearn.metrics import mean_squared_error

print("Mean squared error: ", mean_squared_error(y_test, y_pred))

在上述示例中,我们使用sklearn自带的糖尿病数据集,并使用80%的数据进行训练集和20%的数据进行测试集,创建了一个线性SVM回归器,并使用训练集训练了该回归器。然后我们使用测试集来预测数据点的标签,并输出了均方误差。

相关文章

Python中超好用的split()函数,详解

文章目录一、split函数简介二、语法三、分离字符串四、实例一、split函数简介Python中split()函数,具体作用如下:拆分字符串。通过指定分隔符对字符串进行切片,并返回分割后的字符串列表(...

Python字符串分割,你真的会用split()吗?3分钟掌握核心技巧!

你是否在处理文本数据时被字符串分割搞得头大?是否因为split()的“坑”导致程序报错?今天,一篇搞定Python字符串分割的终极指南,建议收藏!一、快速入门:split()能做什么?假设你有一串文本...

Python分隔字符串函数用法split

aaa,bbb = str.split('&&',2)第一个参数为分隔符第二个参数是要完成的最大拆分数...

站长在线Python精讲使用正则表达式的split()方法分割字符串详解

欢迎你来到站长在线的站长学堂学习Python知识,本文学习的是《在Python中使用正则表达式的split()方法分割字符串详解》。使用正则表达式分割字符串在Python中使用正则表达式的split(...

简单学Python——re库(正则表达式)2(split、findall、和sub)

1、split():分割字符串,返回列表语法:re.split('分隔符','目标字符串')例如:import rere.split(',','...