我对神经网络很陌生,这就是为什么我决定使用Tflearn的原因,因为它非常直观。
但是我找不到我的问题的答案。 tflearn文档提供了以下示例,可让深度神经网络预测某些内容:

network = ...
model = DNN(network)
model.load('model.tflearn')
model.predict(X)


我在网络中插入了一些批处理规范化层,因为我的模型似乎过拟合。 model.predict()是否会自动“告诉”批处理规范化层,使其在训练阶段不表现出来?还是我必须以某种方式指定此
tflearn.config.is_training (is_training=False, session=None)

如果是,您知道我应该把这行放在哪里吗?以及如何创建会话,使其像我的代码一样执行。目前,它基本上看起来像是tflearn.org上的示例:

net = tflearn.input_data(shape=[None, 784])
net = tflearn.fully_connected(net, 64)
net = tflearn.dropout(net, 0.5)
net = tflearn.fully_connected(net, 10, activation='softmax')
net = tflearn.regression(net, optimizer='adam',
loss='categorical_crossentropy')

model = tflearn.DNN(net)
model.fit(X, Y)


除了我使用批处理规范化层,并且使用神经网络进行函数逼近。不幸的是,由于它位于另一台计算机上,所以我现在无法发布我的代码,但实际上基本上是相同的。

有人可以帮我解决这个问题吗?

提前致谢!

最佳答案

在训练和预测时,您需要将tflearn.is_training设置为True或False,tflearn将负责其余的工作。定义模型后,您可以通过以下方法进行训练:

with tf.Session() as sess:
    tflearn.is_training(True, session=sess)
 model.fit(X, Y)


然后预测使用:

with tf.Session as sess:
    tflearn.is_training(False, session=sess)
model.predict(X)

关于python - tflearn.models.dnn.DNN预测时是否会自动关闭辍学层并进行批量归一化?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/51247304/

10-10 16:56