Pattern Classification(1)_perceptron(2)

首先导入产生的数据

pd_data = pd.read_csv('data.csv').ix[:,['x','y']]

方法一

使用

sklearn.linear_model.Perceptron函数

代码如下：

# -*- coding: utf-8 -*-
"""
Created on Tue Oct 25 21:29:20 2016

@author: QING
"""

import pandas as pd
import numpy as np
from pandas import DataFrame,Series
import math
import matplotlib.pyplot as plt

#import data
pd_data = pd.read_csv('data.csv').ix[:,['x','y']]

from sklearn import linear_model
reg = linear_model.Perceptron()
Y = np.append(np.ones(500),np.zeros(500))
X = [pd_data.T[i] for i in range(1000)]
reg.fit(X, Y)
plt.plot(pd_data['x'],pd_data['y'],'.')

plt.plot([-30,50],[(-reg.intercept_-reg.coef_[0][0]*(-30))/reg.coef_[0][1]\
,(-reg.intercept_-reg.coef_[0][0]*50)/reg.coef_[0][1]\
],'r-')