Как читать простой CSV-файл с помощью Datavec

Question

Я хочу прочитать простой 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).

Изменение labelIndex с 0 на 1 приводит к той же ошибке.

Данные в файле выглядят так:

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

Как мне прочитать этот файл? Я предполагаю, что результатом должен быть DataSet, как вход для dl4j.

Answer 1

У вас нет ярлыков. Вам нужно как минимум 2 значения. RecordReaderDataSetIterator обычно ожидает, что где-то там будет метка. Вы не определили ничего, чтобы учиться. Я не уверен, насколько ты новичок в машинном обучении, но я бы отступил назад и сначала изучил основы.