我试图在python 3.7.6上运行这个项目。tflearn
现在我已经安装了Tensorflow 2.0,我想让这些代码使用tf2.0和keras进行编码。
net = tflearn.input_data(shape=[None, len(training[0])])
net = tflearn.fully_connected(net, 8)
net = tflearn.fully_connected(net, 8)
net = tflearn.fully_connected(net, len(output[0]), activation="softmax")
net = tflearn.regression(net)
net = tflearn.DNN(net)
net.fit(training, output, n_epoch=1000, batch_size=8, show_metric=True)
net.save("model.tflearn")
我对tflearn了解不多,但在tensorflow和keras中,它看起来像这样------。
model = tf.keras.models.Sequential([
tf.keras.layers.Dense(8, activation='relu',input_shape=[len(train_dataset.keys())]),
tf.keras.layers.Dense(8, activation='relu'),
tf.keras.layers.Dense(len(output[0], activation='softmax')])
model.compile(optimizer = 'adam,
loss = 'mse',
metrics=['mse'])
model.fit(x, y, epochs=1000)