1. 程式人生 > >關於TensorFlow的MNIST資料集下載指令碼input_data.py的坑

關於TensorFlow的MNIST資料集下載指令碼input_data.py的坑

今天用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檔案,然後就可以搞了。

(要注意資源目錄的問題)