详细讲述支持向量机模型的python代码精确到每一个变量定义
以下是一个简单的支持向量机模型的Python代码,包括变量定义和模型训练:
# 导入必要的库
import numpy as np
from sklearn import svm
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
# 定义数据集
X = np.array([[0, 0], [1, 1], [1, 0], [0, 1]])
y = np.array([0, 1, 1, 0])
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
# 定义支持向量机模型
clf = svm.SVC(kernel='linear', C=1)
# 训练模型
clf.fit(X_train, y_train)
# 预测结果
y_pred = clf.predict(X_test)
# 计算准确率
acc = accuracy_score(y_test, y_pred)
# 输出结果
print("准确率:", acc)
在上述代码中,我们首先导入了必要的库,包括numpy、sklearn的svm模块、train_test_split函数和accuracy_score函数。
然后,我们定义了一个简单的数据集X和对应的标签y。其中,X是一个二维数组,每一行表示一个样本,每个样本有两个特征;y是一个一维数组,每个元素表示对应样本的标签。
接着,我们使用train_test_split函数将数据集划分成训练集和测试集。其中,test_size参数表示测试集占总数据集的比例,random_state参数表示随机种子,保证每次划分的结果一致。
然后,我们定义了一个支持向量机模型clf。其中,kernel参数表示核函数的类型,这里选择了线性核函数;C参数表示正则化强度,这里选择了默认值1。
接下来,我们使用fit函数训练模型,将训练集X_train和y_train作为参数传入。
然后,我们使用predict函数对测试集进行预测,将预测结果存储在y_pred变量中。
最后,我们使用accuracy_score函数计算模型在测试集上的准确率,并将结果输出
原文地址: https://www.cveoy.top/t/topic/honv 著作权归作者所有。请勿转载和采集!