LSM分别苹果和桔子

#!/usr/bin/env python
#-*- coding: utf-8 -*-
import numpy as np
b=1
a=0.2
x = np.array([[1,-1,-1],[1,1,-1]])
d =np.array([-1,1])
w=np.array([0,0,0])
expect_e=0.0005
maxtrycount=20


def sgn(v):
        if v>0:
                return 1
        else:
                return -1
def get_v(myw,myx):
        return (np.dot(myw,myx.T))
def neww(oldw,myd,myx,a):
        mye=get_e(oldw,myx,myd)
        return (oldw+2*a*mye*myx,mye)
def get_e(myw,myx,myd):
        return myd-get_v(myw,myx)




mycount=0
while True:
        mye=0
        i=0          
        for xn in x:
                w,e=neww(w,d[i],xn,a)
                i+=1
                mye+=pow(e,2) 
                print w
                print e
        mye/=float(i)
        mycount+=1
        print u"第 %d 次调整后的权值："%mycount
        print w
        print u"误差：%f"%mye        
        if mye<expect_e or mycount>maxtrycount:break 
               
for xn in x:
        print xn
        print get_v(w,xn)

        print