Lesen von CSV-Dateien mit TensorFlow

Question

Ich versuche, einen Tensor aus einer CSV-Datei zu lesen und zu drucken. Ich folgte dem Rat here, aber das Skript hängt noch. data.csv besteht aus einer Zeile:

1.5,2.5 

Hier ist der Code, der es liest:

datafile = tf.train.string_input_producer([os.path.join(os.getcwd(), "data.csv")]) 
reader = tf.TextLineReader() 
_, value = reader.read(datafile) 
record_defaults = [[1], [1]] 
col1, col2 = tf.decode_csv(value, record_defaults=record_defaults) 
result = tf.stack([col1, col2]) 
config = tf.ConfigProto(inter_op_parallelism_threads=2) 
with tf.Session() as sess: 
    print(sess.run(result)) 

Irgendwelche Gedanken?

Answer 1

Sie vermissen diesen Teil aus der Antwort, die Sie zitiert haben. Wenn Sie den Koordinator nicht hinzufügen und die Warteschlangenlaufwerke nicht starten, wird die Warteschlange nicht in die Warteschlange eingereiht, und die Sitzung hängt ab und wartet darauf, dass ein Element in die Warteschlange eingereiht wird.

with tf.Session() as sess: 
    # Start populating the filename queue. 
    coord = tf.train.Coordinator() 
    threads = tf.train.start_queue_runners(coord=coord) 

    for i in range(1200): 
    # Retrieve a single instance: 
    example, label = sess.run([features, col5]) 

    coord.request_stop() 
    coord.join(threads)