当前位置: 首页 > 工具软件 > Pydata > 使用案例 >

关于TensorFlow的MNIST数据集下载脚本input_data.py的坑

双子民
2023-12-01

今天用github上的代码入门tensorflow但是发现似乎要下载数据集,但是这个我弄了一会才明白是怎么下的,所以把经验写在下面:(ubuntu14.04环境)

用github上的input_data.py


from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import gzip
import os
import tempfile

import numpy
from six.moves import urllib
from six.moves import xrange  # pylint: disable=redefined-builtin
import tensorflow as tf
from tensorflow.contrib.learn.python.learn.datasets.mnist import read_data_sets

把这个文件和github上的nearst_neighbor.py放在一个py工程目录下的两个py文件,然后就可以搞了。

(要注意资源目录的问题)

 类似资料: