как я должен запустить это в tensflow 2.0? - PullRequest
0 голосов
/ 04 мая 2020

Я пытаюсь запустить это на 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")

1 Ответ

0 голосов
/ 05 мая 2020

Я мало что знаю о tflearn, но в тензорном потоке и керасе это будет выглядеть примерно так:

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)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...