1. 程式人生 > >初始化神經網路的權重和偏置

初始化神經網路的權重和偏置

import  numpy as np

#定義網路結構
layers=np.array([4,3,4,2])
'''
第0層  輸入層,維度為4
第一層 4*3 矩陣
第二層 3*4 矩陣
第三層 4*2 矩陣
'''


#權重  範圍設定在-0.25~0.25之間
weights=[]
for i in range(len(layers)-1):
    tmp = (np.random.random([layers[i],layers[i+1]])*2-1)*0.25
    weights.append(tmp)


#bias
'''
第一層  3個
第二層  4個
第三層  2個
'''
bias=[] for i in range(1,len(layers)): tmp = (np.random.random(layers[i])*2-1)*0.25 bias.append(tmp) print weights print bias exit(0)