如何使用 Datavec 读取简单的 CSV 文件

Question

我想使用 Datavec 读取一个简单的 CSV 文件，其中只有一个数字列表，以便在 Deeplearning4j 中使用。 我尝试了许多示例，但不断出现错误。 例如当我执行此操作时：

    RecordReader rrTest = new CSVRecordReader();
    rrTest.initialize(new FileSplit(new File(INPUT_FILE)));
    DataSetIterator testIter = new RecordReaderDataSetIterator(rrTest, 150, 0, 1);

我收到此错误：

Exception in thread "main" org.nd4j.linalg.exception.ND4JIllegalStateException: Invalid shape: Requested INDArray shape [144, 0] contains dimension size values < 1 (all dimensions must be 1 or more).

将 'labelIndex' 从 0 更改为 1 会产生相同的错误。

文件中的数据如下所示：

112
118
132
129
121
135
148
148
136
119
104
118
115

如何阅读此文件？我猜结果应该是一个 DataSet，作为 dl4j 的输入。

Answer 1

你没有标签。您至少需要 2 个值。 RecordReaderDataSetIterator 通常希望在某处有一个标签。你还没有定义要学习的东西。我不确定您对机器学习有多陌生，但我会先退一步学习基础知识。