"Watermelon book," Chapter V, neural network

▶ using a neural network for classification as a casual point

● a single layer perceptron, the code

  1 import numpy as np
  2 import matplotlib.pyplot as plt
  3 from mpl_toolkits.mplot3d import Axes3D
  4 from mpl_toolkits.mplot3d.art3d import Poly3DCollection
  5 from matplotlib.patches import Rectangle
  6 
  7 dataSize = 2000
  8 trainRatio = 0.3
  9 ita = 0.3
 10 epsilon = 0.001
 11 maxTurn = 300
 12 
 13 colors = [[0.5,0.25,0],[1,0,0],[0,0.5,0],[0,0,1],[1,0.5,0]] # 棕红绿蓝橙
14 Trans = 0.5
 15  
16  DEF dataSplit (Data, Part):                                   # The data set is divided into training and test sets 
. 17      return Data [0: Part,:], Data [Part:,:]
 18 is  
. 19  DEF mysign (X) :                                               # distinguish np.sign () 
20 is      return (+ np.sign. 1 (X)) / 2
 21 is  
22 is  DEF function (X, para):                                        # continuous regression function, a drawing 
23 is      return np.sum (X * para [ 0]) - para [. 1]                     # note is minus 
24  
25  DEFJudge (X, para):                                          # classification function, the multiply-add portion and the threshold part 
26 is      return mysign (function (X, para))
 27  
28  DEF createData (Dim, len):                                    # generate test data 
29      np.random.seed (103 )
 30      Output = np.zeros ([len, + Dim. 1 ])
 31 is      for I in (Dim) Range:
 32          Output [:, I] = np.random.rand (len)        
 33 is      Output [:, Dim] List = (Map ( the lambda X: int (X> 0.5), (. 3 - Dim * 2) * Output [:, 0] + 2 * np.sum (Output [:,. 1: Dim],. 1 )))
 34     #print(output, "\n", np.sum(output[:,-1])/len)
 35     return output   
 36 
 37 def perceptron(data):                                       # 单层感知机
 38     len = np.shape(data)[0]
 39     dim = np.shape(data)[1] - 1    
 40     xE = np.concatenate((data[:, 0:-1], -np.ones(len)[:,np.newaxis]), axis = 1)
 41     w = np.zeros(dim + 1)
 42     finishFlag = False    
 43     count = 0
 44     while finishFlag == False and count < maxTurn:        
 45         error = 0.0
 46         for i in range(len):            
 47             delta = ita * (data[i,-1] - mySign(np.sum(xE[i] * w))) * xE[i]   # 注意 np.sign 的定义
 48             error += np.sum(delta * delta)                
 49             w += delta
 50         count += 1        
 51         print("count = ",count,", w = ", w, ", error = ", error)
 52         if error < epsilon:
 53             finishFlag = True
 54             break            
 55     return (w[:-1],w[-1])
 56 
 57 def test(dim):                                              # 测试函数
 58     allData = createData(dim, dataSize)
 59     trainData, testData = dataSplit(allData, int(dataSize * trainRatio))
 60 
 61     para = perceptron(trainData)
 62     
 63     myResult = [ judge(i[0:dim], para) for i in testData ]   
 64     = np.sum errorRatio ((np.array (The myResult) - testData [:, -. 1] .astype (int)) ** 2) / (dataSize * (l- trainRatio))
 65      Print ( " Dim = " + STR (Dim) + " , errorRatio = " + STR (round (errorRatio,. 4 )))
 66      IF Dim> =. 4:                                             # . 4 dimension than does drawing, only the output of the error ratio 
67          return 
68  
69      errorP = []                                              # drawing section , the test data set into error classes, class 0 and class 1 
70      Class1 = []
 71 is      Class0 = []
 72      for i in range(np.shape(testData)[0]):
 73         if myResult[i] != testData[i,-1]:
 74             errorP.append(testData[i])
 75         elif myResult[i] == 1:
 76             class1.append(testData[i])
 77         else:
 78             class0.append(testData[i])
 79     errorP = np.array(errorP)
 80     class1 = np.array(class1)
 81     class0 = np.array(class0)
 82 
 83     fig = plt.figure(figsize=(10, 8))                
 84     
 85     if dim == 1:
 86         plt.xlim(0.0,1.0)
 87         plt.ylim(-0.25,1.25)
 88         plt.plot([0.5, 0.5], [-0.5, 1.25], color = colors[0],label = "realBoundary")                
 89         plt.plot([0, 1], [ function(i, para) for i in [0,1] ],color = colors[4], label = "myF")
 90         plt.scatter(class1[:,0], class1[:,1],color = colors[1], s = 2,label = "class1Data")                
 91         plt.scatter(class0[:,0], class0[:,1],color = colors[2], s = 2,label = "class0Data")                
 92         if len(errorP) != 0:
 93             plt.scatter(errorP[:,0], errorP[:,1],color = colors[3], s = 16,label = "errorData")        
 94         plt.text(0.2, 1.12, "realBoundary: 2x = 1\nmyF(x) = " + str(round(para[0][0],2)) + " x - " + str(round(para[1],2)) + "\n errorRatio = " + str(round(errorRatio,4)),\
 95             size=15, ha="center", va="center", bbox=dict(boxstyle="round", ec=(1., 0.5, 0.5), fc=(1., 1., 1.)))
 96         R = [Rectangle((0,0),0,0, color = colors[k]) for k in range(5)]
 97         plt.legend(R, ["realBoundary", "class1Data", "class0Data", "errorData", "myF"], loc=[0.81, 0.2], ncol=1, numpoints=1, framealpha = 1)        
 98     
 99     if dim == 2:        
100         plt.xlim(0.0,1.0)
101         plt.ylim(0.0,1.0)
102         plt.plot([0,1], [0.25,0.75], color = colors[0],label = "realBoundary")        
103         xx = np.arange(0, 1 + 0.1, 0.1)                
104         X,Y = np.meshgrid(xx, xx)
105         contour = plt.contour(X, Y, [ [ function((X[i,j],Y[i,j]), para) for j in range(11)] for i in range(11) ])
106         plt.clabel(contour, fontsize = 10,colors='k')
107         plt.scatter(class1[:,0], class1[:,1],color = colors[1], s = 2,label = "class1Data")        
108         plt.scatter(class0[:,0], class0[:,1],color = colors[2], s = 2,label = "class0Data")        
109         if len(errorP) != 0:
110             plt.scatter(errorP[:,0], errorP[:,1],color = colors[3], s = 8,label = "errorData")        
111         plt.text(0.75, 0.92, "realBoundary: -x + 2y = 1\nmyF(x,y) = " + str(round(para[0][0],2)) + " x + " + str(round(para[0][1],2)) + " y - " + str(round(para[1],2)) + "\n errorRatio = " + str(round(errorRatio,4)), \
112             size = 15, ha="center", va="center", bbox=dict(boxstyle="round", ec=(1., 0.5, 0.5), fc=(1., 1., 1.)))
113         R = [Rectangle((0,0),0,0, color = colors[k]) for k in range(4)]
114         plt.legend(R, ["realBoundary", "class1Data", "class0Data", "errorData"], loc=[0.81, 0.2], ncol=1, numpoints=1, framealpha = 1)     
115 
116     if dim == 3:        
117         ax = Axes3D(fig)
118         ax.set_xlim3d(0.0, 1.0)
119         ax.set_ylim3d(0.0, 1.0)
120         ax.set_zlim3d(0.0, 1.0)
121         ax.set_xlabel('X', fontdict={'size': 15, 'color': 'k'})
122         ax.set_ylabel('Y', fontdict={'size': 15, 'color': 'k'})
123         ax.set_zlabel('W', fontdict={'size': 15, 'color': 'k'})
124         v = [(0, 0, 0.25), (0, 0.25, 0), (0.5, 1, 0), (1, 1, 0.75), (1, 0.75, 1), (0.5, 0, 1)]
125         f = [[0,1,2,3,4,5]]
126         poly3d = [[v[i] for i in j] for j in f]
127         ax.add_collection3d(Poly3DCollection(poly3d, edgecolor = 'k', facecolors = colors[0]+[trans], linewidths=1))        
128         ax.scatter(class1[:,0], class1[:,1],class1[:,2], color = colors[1], s = 2, label = "class1")                       
129         ax.scatter(class0[:,0], class0[:,1],class0[:,2], color = colors[2], s = 2, label = "class0")                       
130         if len(errorP) != 0:
131             ax.scatter(errorP[:,0], errorP[:,1],errorP[:,2], color = colors[3], s = 8, label = "errorData")                
132         ax.text3D(0.8, 1, 1.15, "realBoundary: -3x + 2y +2z = 1\nmyF(x,y,z) = " + str(round(para[0][0],2)) + " x + " + \
133             str(round(para[0][1],2)) + " y + " + str(round(para[0][2],2)) + " z - " + str(round(para[1],2)) + "\n errorRatio = " + str(round(errorRatio,4)), \
134             size = 12, ha="center", va="center", bbox=dict(boxstyle="round", ec=(1, 0.5, 0.5), fc=(1, 1, 1)))
135         R = [Rectangle((0,0),0,0, color = colors[k]) for k in range(4)]
136         plt.legend(R, ["realBoundary", "class1Data", "class0Data", "errorData"], loc=[0.83, 0.1], ncol=1, numpoints=1, framealpha = 1)
137         
138     fig.savefig("R:\\dim" + str(dim) + ".png")
139     plt.close()        
140 
141 if __name__ == '__main__':
142     test(1)
143     test(2)    
144     test(3)
145     test(4)

● output, a one-dimensional convergence 6 times, 39 times the two-dimensional, three times 87, 89 over the four-dimensional

count =  1 , w =  [1.39770297 0.45      ] , error =  5.222155402906847
count =  2 , w =  [1.52702391 0.75      ] , error =  1.6525212964325524
count =  3 , w =  [1.99396967 0.75      ] , error =  3.8768795270185015
count =  4 , w =  [2.0734924 1.05     ] , error =  2.1108010510371176
count =  5 , w =  [2.08542138 1.05      ] , error =  0.2226526876961404
count =  6 , w =  [2.08542138 1.05      ] , error =  0.0
dim = 1, errorRatio = 0.0043
count =  1 , w =  [-0.86290479  1.73543324  0.45      ] , error =  10.072859142827403
count =  2 , w =  [-1.13927193  2.04084504  0.45      ] , error =  4.4609363561371165
count =  3 , w =  [-1.15553275  2.73079122  0.75      ] , error =  7.930218436293679

...

count =  37 , w =  [-3.24442847  6.45406089  1.65      ] , error =  0.5427638905658313
count =  38 , w =  [-3.25333719  6.53959919  1.65      ] , error =  2.6670168284877542
count =  39 , w =  [-3.25333719  6.53959919  1.65      ] , error =  0.0
dim = 2, errorRatio = 0.0
count =  1 , w =  [-1.83648172  1.13115561  1.48731853  0.15      ] , error =  15.299397796880317
count =  2 , w =  [-2.21432213  1.4718464   1.61271137  0.45      ] , error =  5.514265592439243
count =  3 , w =  [-2.65894841  1.73243095  1.7833203   0.45      ] , error =  5.782491180281051

...

count =  84 , w =  [-8.76231537  5.53262355  5.91015865  1.35      ] , error =  2.3909384841900865
count =  85 , w =  [-8.80616091  5.60771114  5.82973106  1.35      ] , error =  1.9246305102403725
count =  86 , w =  [-8.77906986  5.58426138  5.92865995  1.35      ] , error =  0.4663079739497135
count =  87 , w =  [-8.77906986  5.58426138  5.92865995  1.35      ] , error =  0.0
dim = 3, errorRatio = 0.0121
count =  1 , w =  [-2.55073743  0.77544673  0.79572989  1.11402485  0.15      ] , error =  21.2911454497099
count =  2 , w =  [-3.17104932  1.33820515  1.13266849  1.11805123  0.15      ] , error =  11.542535217905032
count =  3 , w =  [-3.56132108  1.37329683  1.34578486  1.58997522  0.45      ] , error =  9.590714392019622

...

count =  87 , w =  [-10.13469991   4.36826988   4.06767039   4.25598478   0.75      ] , error =  1.9723755293196636
count =  88 , w =  [-10.27022167   4.12906605   4.11467555   4.10943389   1.05      ] , error =  1.5046863818100382
count =  89 , w =  [-10.27022167   4.12906605   4.11467555   4.10943389   1.05      ] , error =  0.0
dim = 4, errorRatio = 0.0014

● Paint

Guess you like

Origin www.cnblogs.com/cuancuancuanhao/p/11128493.html