neural-network - DeepLearning4J - 获取数据和训练模型

Question

我尝试创建最简单的 NeuralNetwork 并使用一些数据对其进行训练：因此我创建了一个具有以下模式的 test.csv：

数字，数字+1；

数字2，数字2+1

...

我尝试用网络进行线性回归......

但是我没有找到获取数据的方法，DataSetIterator 不起作用。

如何拟合数据，如何测试数据？

Answer 1

在我们的示例中，我们鼓励人们使用 datavec + recordreaderdatasetiterator。

Datavec 具有所有各种数据加载组件。在没有看到任何代码的情况下，我不确定您所说的“datasetiterator not working”是什么意思，但您似乎并没有真正看过我们的示例。

在 csv 记录阅读器的多个示例中，您可以将其用于回归和分类用例。考虑重新定位您的数据管道以使用它们。这些示例总是在这里找到： https ://github.com/deeplearning4j/dl4j-examples

如果您遵循其中任何一个，就会出现相同的模式：Record reader for any data format -> RecordReaderDataSetIterator

迭代器允许您指定常见的构造函数，例如它是否是回归，您的标签是哪一列等。