Почему Keras model.predict никогда не выполняется на этой простой модели машинного обучения, которую я настроил?

Question

import numpy as np
import keras as ks

model = ks.Sequential([ks.layers.Dense(units=1, input_shape=[1])])
model.compile(optimizer='sgd', loss='mean_squared_error')

xs=np.array([-1.0, 0.0, 1.0, 2.0, 3.0, 4.0], dtype = float)
ys=np.array([-3.0, -1.0, 1.0, 3.0, 5.0, 7.0], dtype = float)

model.fit(xs, ys, epochs=500)

print(model.predict([10.0]))

model.predict никогда не запускается независимо от того, как долго я жду. Какие тесты я могу сделать, чтобы диагностировать проблему? Какие вопросы я должен задать себе, чтобы найти проблему?

Answer 1

Я вернулся и исправил свою сборку?

Сначала я переустановил тензор потока. Моя первоначальная команда была:

pip install tensorflow-gpu==2.0

, которую я установил, используя следующее:

pip install tensorflow

И затем я убедился, что выполнил следующие шаги для Windows.

https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html

Answer 2

Код в порядке - для меня менее чем за 1 секунду.

6/6 [==============================] - 0s 101us/step - loss: 4.8739e-05
Epoch 500/500

6/6 [==============================] - 0s 115us/step - loss: 4.7737e-05
[[18.979841]]

Process finished with exit code 0