python - 将 TensorFlow 1.0 代码转换为 2.0 版本

Question

我在将 tensorflow 1.0 代码转换为 2.0 时遇到问题

我可以成功转换这个

版本 1

 # reset underlying graph data
 #tf.reset_default_graph()

版本 2

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

这是我遇到问题的下面的代码，请帮助我如何在 tensorflow 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')

标签： pythontensorflow

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）