python SVM调包线性分类西瓜

最新推荐文章于 2022-12-13 23:30:00 发布

wwxy261

最新推荐文章于 2022-12-13 23:30:00 发布

阅读量1.3k

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/wwxy1995/article/details/79782733

版权

算法专栏收录该内容

3633 篇文章 120 订阅

订阅专栏

import xlrd
import numpy as np
from sklearn.model_selection import KFold
from sklearn.model_selection import cross_val_score
from sklearn.svm import SVC

data = xlrd.open_workbook('gua.xlsx')
sheet = data.sheet_by_index(0)
Density = sheet.col_values(6)
Sugar = sheet.col_values(7)
Res = sheet.col_values(8)

# 读取原始数据
X = np.array([Density, Sugar])
# y的尺寸为(17,)
y = np.array(Res)
X = X.reshape(17, 2)
num_folds = 10
seed = 7
kfold = KFold(n_splits=num_folds, random_state=seed)
model = SVC()
result = cross_val_score(model, X, y, cv=kfold)
print(result.mean())