如何使用Datavec

时间:2018-02-07 06:59:49

标签: csv deeplearning4j recordreader

我想阅读一个简单的CSV文件,其中只包含使用Datavec的数字列表,以便在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).

更改&#39; labelIndex&#39;从0到1给出了相同的错误。

文件中的数据如下所示:

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

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

1 个答案:

答案 0 :(得分:0)

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