【发布时间】:2019-06-18 08:08:23
【问题描述】:
我正在尝试从包含所需网络结构的列表中使用 Tensorflow 在 for 循环中创建一个简单的 MLP,例如结构 = [100, 50, 20, 1]。在这个列表中,100 代表输入大小,1 代表输出大小。 (我需要这个用于预测应用程序,但这与我的问题并不严格相关。)
我还没有看到在 Tensorflow 中创建网络的类似方法。出于部分逃避我的原因,人们似乎建议最好单独声明每个变量,例如layer_1 = x1 * w1 + b1 然后 layer_2 = x2 * w2 + b2。创建我在 for 循环 [for i in range(len(structure)-1):] 中使用的网络的动态方法是否错误?对我来说,网络似乎工作正常,张量板上显示的网络结构似乎是正确的。
您认为这种创建网络的方式好吗?您是否认为我在不知不觉中陷入了任何 Tensorflow / Context Manager 问题?
import tensorflow as tf
class Model(object):
def __init__(self, structure, lr=0.01):
assert structure[-1] == 1
input_size = structure[0]
act_fun = tf.nn.tanh
G = tf.Graph()
with G.as_default():
self.X = tf.placeholder(tf.float32, shape=[None, input_size])
self.Y = tf.placeholder(tf.float32, shape=[None, 1])
X_out = self.X
for i in range(len(structure)-1):
from_, to = structure[i], structure[i+1]
initializer = tf.variance_scaling_initializer()
w = tf.Variable(initializer([from_, to]), dtype=tf.float32, name=f'W{i}')
b = tf.Variable(tf.zeros(to), name=f'B{i}')
if to != 1:
X_out = act_fun(tf.matmul(X_out, w) + b)
else:
X_out = tf.matmul(X_out, w) + b
self.forecast_layer = X_out
self.loss = tf.losses.mean_squared_error(self.Y, self.forecast_layer)
self.trainer = tf.train.GradientDescentOptimizer(lr).minimize(self.loss)
self.init = tf.global_variables_initializer()
self.session = tf.Session(graph=G)
self.session.run(self.init)
self.session.graph.finalize()
def fit(self, X, Y):
self.session.run(self.trainer, feed_dict={self.X:X, self.Y:Y})
def forecast(self, X):
return self.forecast_layer.eval(feed_dict={self.X:X}, session=self.session)
def evaluate_loss(self, X, Y):
return self.loss.eval(feed_dict={self.Y:Y, self.forecast_layer:self.forecast(X)}, session=self.session)
M = Model([100, 50, 20, 1], lr=0.001)
【问题讨论】:
-
我不认为人们想“建议最好”单独创建每一层,只是大多数示例代码要么真的很糟糕/过时,要么他们只想明确地创建每一层(循环可以隐藏网络的复杂性/大小)。循环版本更加灵活,因此您绝对应该喜欢它。
标签: tensorflow neural-network forecasting