首页 文章

如何在tensorflow程序中加载csv文件?

提问于
浏览
0

我正在通过卷积神经网络的tensorflow教程研究深度神经网络 . 但我拥有的数据集是csv文件的形式 . 我有4个csv文件2用于培训(xtrain.csv和ytrain.csv),2个用于测试(xtest.csv和ytest.csv) . xtrain.csv和xtest.csv包含分别具有1280行和4096列以及320行和4096列的输入 . 这里每行代表尺寸1x4096的图像 . 因此输入层将有4096个神经元 . 现在,ytrain.csv和ytest.csv包含分别具有32行和1280列以及32行320列的输出,其中每列表示以一个热矢量形式的特定图像的输出 . 所以输出层中会有32个神经元 . 有人可以指导我如何将输入和输出(标签)的csv文件加载到我的程序中吗?

1 回答

  • 0

    Tensorflow的功能如下所示 .

    tf.decode_csv(records, record_defaults, field_delim=None, name=None)
    

    您可以使用它将数据加载到张量中 . 从那里你只需要将张量转换为正确的形状 .

    imgs = tf.decode_csv(csv_file, tf.float32)
    imgs = tf.reshape(imgs,shape=(num_of_imgs,64,64))
    

    请记住,tf.decode_csv每列都将是一个张量 . 你可以为标签做类似的事情 .

相关问题