關於TensorFlow的MNIST資料集下載指令碼input_data.py的坑
阿新 • • 發佈:2019-01-06
今天用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檔案,然後就可以搞了。
(要注意資源目錄的問題)