【问题标题】:Train and test in TensorFlow with CSV files使用 CSV 文件在 TensorFlow 中进行训练和测试
【发布时间】:2017-04-08 15:35:21
【问题描述】:

我有一个 train.csv 和一个 test.csv。我想使用 tensorflow 查看 training.csv 并为 test.csv 文件打印出两列 CSV 文件。第一列是 id,第二列是 tensorflow 预测的 id 类别。我正在使用 python。

【问题讨论】:

  • 你的问题是什么? (提示:隐含的“有人会为我写这个吗?”的答案通常是否定的)

标签: python python-3.x csv tensorflow


【解决方案1】:

无论 Tensorflow 预测什么,一旦您对其执行 .eval(),它就会成为“标准”Python 数据类型,您可以使用通常的非 Tensorflow API 将其写入 CSV。

这是一个可以做到这一点的代码(从一个较大文件的各个部分拼凑而成,如果不完全连贯,请原谅)

lVerb = open('your_file.csv', 'w')
lVerb.write('Predcited,Actual\n')

guessY = tf.argmax(Y, 1).eval({X: feature_batch, keep_prob: 1})
trueY = tf.argmax(label_batch, 1).eval()

for j in range(0, batch_size):
    curGuess = guessY[j]
    curTrue = trueY[j]
    lVerb.write('{},{}\n'.format(curGuess, curTrue))

lVerb.close()

正如您在上面看到的,TF 的 .eval() 操作或 Session.run() 的输出是标准 Python 数据类型,例如 numpy 数组/等,具体取决于您在做什么。在上述情况下,您将它们独立于 Tensorflow 写入 CSV

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2021-04-19
    • 2020-06-17
    • 2019-07-22
    • 2017-09-27
    • 2020-02-28
    • 2019-05-01
    相关资源
    最近更新 更多