SVM算法进行二分类
·
1.预测类别是1还是2
import numpy as np X = np.array([[-1, 1], [-2, -1], [3, 1], [2, 1]]) #数据特征 y = np.array([1, 2, 1, 2]) # 数据对应的标签 from sklearn.svm import SVC # 导入svm的svc类(支持向量分类) clf = SVC() # 创建分类器对象 clf.fit(X, y) # 用训练数据拟合分类器模型 SVC(C=1.0, cache_size=200) # 用训练好的分类器去预测[0.8, -1]数据的标签[1] print(clf.predict([[0.8, -1]]))
输出结果:[2]
(表明是类别2)
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)