天天看點

TensorFlow實戰-多層感覺機

# 建立一個Tensorflow預設的Interactive Session,這樣後面執行各項操作就無須指定Session了      
# 将權重初始化為截斷的正态分布,其标準差為0.1
in_units = 784
h1_units = 300
w1      
# keep_prob訓練時小于1,以制造随機性,防止過拟合。預測時,等于1,即用全部特征來預測樣本的類别。
hidden1_drop = tf.nn.dropout(hidden1, keep_prob)