使用python中的SVM进行数据分类预测

在Python中使用支持向量机(SVM)进行数据分类预测的示例代码如下:

# 引入需要的库
from sklearn.svm import SVC
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score

# 准备数据集,假设有一个特征矩阵X和标签向量Y
# X的每一行是一个样本的特征向量,Y的每个元素对应X对应样本的标签

# 假设特征矩阵X为 1000x3 的数据,标签向量Y为 1000x1 的数据
X = [[0, 0, 0], [0, 1, 1], [1, 0, 1], [1, 1, 0]]
Y = [0, 1, 1, 0]

# 拆分数据集为训练集和测试集,这里按照 70% 的比例划分数据
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.3, random_state=1)

# 建立支持向量机(SVM)模型并设置参数
svm = SVC(kernel='linear')  # 选择线性核函数

# 训练支持向量机模型
svm.fit(X_train, Y_train)

# 使用模型进行预测
Y_test_predicted = svm.predict(X_test)

# 评估模型性能
accuracy = accuracy_score(Y_test, Y_test_predicted)
print("Accuracy:", accuracy)

# 可以根据需要进行模型调参和优化,例如选择其他核函数、调整正则化参数C等

该示例代码使用了sklearn库中的SVC类来构建支持向量机(SVM)模型。首先,将数据集划分为训练集和测试集;然后创建一个SVM模型,设置参数,如选择线性核函数;接下来使用训练集对模型进行训练;最后,利用测试集数据进行预测,并根据预测结果评估模型性能。

根据您的数据集和问题,可以根据需要进行模型参数调整和优化,例如选择其他核函数、调整正则化参数C等,以获得更好的预测结果和性能。

猜你喜欢

转载自blog.csdn.net/weixin_44463965/article/details/131674595