机器学习(5)--创建线性分类器(linear classifier)

创建一个简单的线性分类器

创建分类器代码如下:

#-*- coding:utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt

#创建样本数据
x=np.array([[3,1],[2,5],[1,8],[6,4],[5,2],[3,5],[4,7],[4,-1]])

#为数据点创建标记
y=[0,1,1,0,0,1,1,0]

#将样本数据分成两类 其中range函数指的是x样本数据的长度
class_0=np.array([x[i] for i in range(len(x)) if y[i]==0])
class_1=np.array([x[i] for i in range(len(x)) if y[i]==1])

line_x=range(10)
line_y=line_x

#数据可视化
plt.figure()
#绘制散点图
plt.scatter(class_0[:,0],class_0[:,1],color='black',marker='s')
plt.scatter(class_1[:,0],class_1[:,1],color='black',marker='x')
plt.plot(line_x,line_y,color='black',linewidth=3)

plt.show()

效果图如下所示: