def model(data_feed):
h1 = f(tf.matmul(data_feed, w1) + b1)
h1 = tf.cond(is_training, lambda: tf.nn.dropout(h1, p), lambda: h1)
h2 = f(tf.matmul(h1, w2) + b2)
h2 = tf.cond(is_training, lambda: tf.nn.dropout(h2, p), lambda: h2)
return tf.matmul(h2, w_out) + b_out
评论列表
文章目录