将 Tensorflow 1.0 代码转换为 2.0 版本

Question

我在将我的张量流 1.0 代码转换为 2.0 时遇到问题

我可以转换成功

版本 1

# reset underlying graph data
#tf.reset_default_graph()

版本 2

from tensorflow.python.framework import ops
ops.reset_default_graph()

这是我遇到问题的以下代码，请帮助我如何在张量流 2 版本中构建 NN

。

# Build neural network
net = tflearn.input_data(shape=[None, len(train_x[0])])
net = tflearn.fully_connected(net, 8)
net = tflearn.fully_connected(net, 8)
net = tflearn.fully_connected(net, len(train_y[0]), activation='softmax')
net = tflearn.regression(net)
# Define model and setup tensorboard
model = tflearn.DNN(net, tensorboard_dir='tflearn_logs')
# Start training (apply gradient descent algorithm)
model.fit(train_x, train_y, n_epoch=100, batch_size=8, show_metric=True)
model.save('my_drive/AI_values/model/model.ckpt')

Answer 1

tf_upgrade_v2 —infile foo.py — outfile foo-upgraded.py

此命令可以轻松地将您的 TensorFlow 代码从 1.0 转换为 2.0，以获得您可以参考的完整内容(https://www.youtube.com/watch?v=JmSNUeBG-PQ(

Answer 2

我无法正确理解您的问题(即什么是版本 1 和版本 2，以及您想从 tf 1.0 转换为 tf 2.0 的内容(，但是将任何代码从 tensorflow 版本 1.0 转换为 2.0 非常容易。

在导入张量流而不是执行import tensorflow as tf时，可以执行import tensorflow.compat.v1 as tf，并且如果要从tensorflow导入特定模块(例如from tensorflow.keras.models import Model(，请执行from tensorflow.compat.v1.keras.models import Model。