程序
import cv2
import numpy as np
import matplotlib as plt
#1.准备data
rand1 = np.array([[155,48],[159,50],[164,53],[168,56],[172,60]]) #女生数据
rand2 = np.array([[152,53],[156,55],[160,56],[172,64],[176,65]]) #男生数据
#2.准备label
label = np.array([[0],[0],[0],[0],[0],[1],[1],[1],[1],[1]])
#3 data
data = np.vstack((rand1,rand2))
data = np.array(data,dtype='float32')
#SVM 所有数据都要有标签,表明当前数据是什么类型(男生还是女生)
#监督学习->有标签 0->负样本 1->正样本
#4.训练SVM
svm = cv2.ml.SVM_create() #ml->机器学习模块 SVM_create()->创建svm
#属性设置
svm.setType(cv2.ml.SVM_C_SVC)
svm.setKernel(cv2.ml.SVM_LINEAR) #线性分类器
svm.setC(0.01)
#训练
result = svm.train(data,cv2.ml.ROW_SAMPLE,label)
#预测
pt_data = np.vstack([[167,55],[162,57]]) #0,1
pt_data = np.array(pt_data,dtype='float32')
#print(pt_data)
(par1,par2) = svm.predict(pt_data)
#可视化
plt.scatter(rand1[:,0], rand1[:,1], 80, 'g', 'o')
plt.scatter(rand2[:,0],rand2[:,1], 80, 'b', 's')
plt.scatter(pt_data[:,0], pt_data[:,1], 80, 'r', '*')
plt.show()
print(par1,par2)
出现错误
plt.scatter(rand1[:,0], rand1[:,1], 80, 'g', 'o')
AttributeError: module 'matplotlib' has no attribute 'scatter'
原因:
导入包时用的是
import matplotlib as plt
改为
import matplotlib.pyplot as plt