2017-02-26 5 views
6

私はpython & tensorFlowを使い慣れていません。テンソルフローのドキュメントにはthis MNIST tutorialと書かれています。FLAGS =意味はありませんか?

最初のビットでは、私はFLAGS = Noneがここで何をするのか分かりません。私はGoogleで検索し、空に戻った。これは他人にはあまりにも明白ですか?

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

import argparse 
import sys 

from tensorflow.examples.tutorials.mnist import input_data 

import tensorflow as tf 

FLAGS = None 


def main(_): 
    # Import data 
    mnist = input_data.read_data_sets(FLAGS.data_dir, one_hot=True) 

FLAGSとはどのように使用されていますか? 例:FLAGS.data_dir

ご協力いただけると助かります。

答えて

3

FLAGS=Noneを初期化するだけの方法ですグローバル定数を初期化する。そのまま残すとには属性がないため、mainにエラーが発生します。

しかし、完全な例に示すように​​parserで設定すると、さまざまな属性を持つ単純なオブジェクトになります。 mainは、これらの属性の1つをdata_dirとします。

FLAGS, unparsed = parser.parse_known_args() 
print(FLAGS) 

後にあなたがdata_dirの値は、コマンドラインから解析されたNamespace(data_dir='a directory', ....)を、確認する必要があります。

4

これは、あなたが見ている完全なコードだった:私は説明します:

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

import argparse 
import sys 

from tensorflow.examples.tutorials.mnist import input_data 

import tensorflow as tf 

FLAGS = None #Adds a default value to FLAGS 


def main(_): #Everything inside the function is not checked until it's called 
    mnist = input_data.read_data_sets(FLAGS.data_dir, one_hot=True) #FLAGS is not None anymore because it got changed below 

    x = tf.placeholder(tf.float32, [None, 784]) 
    W = tf.Variable(tf.zeros([784, 10])) 
    b = tf.Variable(tf.zeros([10])) 
    y = tf.matmul(x, W) + b 

    y_ = tf.placeholder(tf.float32, [None, 10]) 

    cross_entropy = tf.reduce_mean(
     tf.nn.softmax_cross_entropy_with_logits(labels=y_, logits=y)) 
    train_step = tf.train.GradientDescentOptimizer(0.5).minimize(cross_entropy) 

    sess = tf.InteractiveSession() 
    tf.global_variables_initializer().run() 
    # Train 
    for _ in range(1000): 
    batch_xs, batch_ys = mnist.train.next_batch(100) 
    sess.run(train_step, feed_dict={x: batch_xs, y_: batch_ys}) 

    # Test trained model 
    correct_prediction = tf.equal(tf.argmax(y, 1), tf.argmax(y_, 1)) 
    accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32)) 
    print(sess.run(accuracy, feed_dict={x: mnist.test.images, 
             y_: mnist.test.labels})) 

if __name__ == '__main__': 
    parser = argparse.ArgumentParser() 
    parser.add_argument('--data_dir', type=str, default='/tmp/tensorflow/mnist/input_data', 
         help='Directory for storing input data') 

    FLAGS, unparsed = parser.parse_known_args() #Here it changed the value of FLAGS to the first thing returned from parser.parse_known_args() 

    tf.app.run(main=main, argv=[sys.argv[0]] + unparsed) #runs the app (calling main) 

起こっていたことは、FLAGSが、ここで変更してしまったということです。 FLAGS, unparsed = parser.parse_known_args()

関連する問題