从CSV文件加载数据
本页将介绍如何把来自CSV文件的数据加载到神经网络中,这是很有用的信息,尤其是在处理时间序列数据时。Deeplearning4j中有一种比较简单的方法:
public static void main(String[] args) throws Exception {
RecordReader recordReader = new CSVRecordReader(0,",");
recordReader.initialize(new FileSplit(new ClassPathResource("iris.txt").getFile()));
//读取器、标签索引、可能的标签数量
DataSetIterator iterator = new RecordReaderDataSetIterator(recordReader,4,3);
//用记录读取器读取数据集信息。datasetiterator负责向量化
DataSet next = iterator.next();
//自定义参数
Nd4j.MAX_SLICES_TO_PRINT = 10;
Nd4j.MAX_ELEMENTS_PER_SLICE = 10;
然后就可以配置神经网络,用这个数据集来定型。具体方法参见示例。
CSVRecordReader类的详情参见此处。