欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

SVM向量机

程序员文章站 2023-12-25 14:15:45
...

svm可以实现分类与回归

使用库sklearn.svm

1 分类
参数kernel 分类类型,默认’rbf’为最好的分类方式,但也要依据实际情况,比如线性分类使用’linear’

svm.SVC()

2 回归
参数kernel 回归类型,默认’rbf’为最好的回归方式,但也要依据实际情况,比如线性回归使用’linear’

svm.SVR(kernel='rbf')

训练
数据要求,x为二维数组,y为1维数组

clt = svm.SVR(kernel='linear')
clt.fit(x, y)

预测
数据要求,x为二维数组,预测结果为1维数组,数组只有一个数,使用[0]取出

clt.predict(x)
相关标签: 数据分析

上一篇:

下一篇: