为什么Tensorflow的教程不适用于PyCharm?

时间:2019-12-14 15:57:46

标签: tensorflow pycharm

我正在尝试学习Tensorflow的基础知识,并且作为开发环境,我正在使用PyCharm。我使用PyCharm创建了一个新项目,下载了tensorflow软件包,并将第一个教程复制并粘贴到了文件中。但是,PyCharm会打印出运行时错误。

我的完整代码是:

n = len(a)
mask = np.zeros(n, dtype=bool)
for i in range(n-1):
    mask[i+1:] |= (np.abs(a[i+1:]-a[i])<=1).all(1)
out = a[~mask]

结果是

import tensorflow as tf

mnist = tf.keras.datasets.mnist

(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0

model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(input_shape=(28, 28)),
  tf.keras.layers.Dense(128, activation='relu'),
  tf.keras.layers.Dropout(0.2),
  tf.keras.layers.Dense(10, activation='softmax')
])

model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

model.fit(x_train, y_train, epochs=5)

model.evaluate(x_test,  y_test, verbose=2)

1 个答案:

答案 0 :(得分:1)

听起来像是访问mnist数据集的问题。

我不确定您使用的是哪个tensorflow版本,但首先要检查tensorflow版本上的文档以下载mnist数据集。

通过其他软件包下载mnist。可以通过sklearn和其他许多方式。以下是一个示例: https://github.com/datapythonista/mnist

import mnist

train_images = mnist.train_images()
train_labels = mnist.train_labels()

test_images = mnist.test_images()
test_labels = mnist.test_labels()