数据可视化 Parameters: dataMat - 数据矩阵 labelMat - 数据标签 Returns: 无
(dataMat, labelMat)
| 31 | return datMat,classLabels |
| 32 | |
| 33 | def showDataSet(dataMat, labelMat): |
| 34 | """ |
| 35 | 数据可视化 |
| 36 | Parameters: |
| 37 | dataMat - 数据矩阵 |
| 38 | labelMat - 数据标签 |
| 39 | Returns: |
| 40 | 无 |
| 41 | """ |
| 42 | data_plus = [] #正样本 |
| 43 | data_minus = [] #负样本 |
| 44 | for i in range(len(dataMat)): |
| 45 | if labelMat[i] > 0: |
| 46 | data_plus.append(dataMat[i]) |
| 47 | else: |
| 48 | data_minus.append(dataMat[i]) |
| 49 | data_plus_np = np.array(data_plus) #转换为numpy矩阵 |
| 50 | data_minus_np = np.array(data_minus) #转换为numpy矩阵 |
| 51 | plt.scatter(np.transpose(data_plus_np)[0], np.transpose(data_plus_np)[1]) #正样本散点图 |
| 52 | plt.scatter(np.transpose(data_minus_np)[0], np.transpose(data_minus_np)[1]) #负样本散点图 |
| 53 | plt.show() |
| 54 | |
| 55 | def stumpClassify(dataMatrix,dimen,threshVal,threshIneq): |
| 56 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected