java - ND4J DL4J 将数据放入 fit 方法
问题描述
我有一个 INDArray 的数据,它是 x 的返回值,如下所示:
private static INDArray createDataSet(String path)throws Exception {
List<String> lines = IOUtils.readLines(new FileInputStream(path), StandardCharsets.UTF_8);
double[] position = new double[lines.size()];
double[] year = new double[lines.size()];
double[] month = new double[lines.size()];
double[] day = new double[lines.size()];
double[] close = new double[lines.size()];
int linecount = 0;
Iterator<String> it = lines.iterator();
while(it.hasNext()) {
String line = it.next();
String[] parts = line.split(",");
position[linecount] = linecount;
year[linecount] = Double.valueOf(parts[0]);
month[linecount] = Double.valueOf(parts[1]);
day[linecount] = Double.valueOf(parts[2]);
close[linecount] = Double.valueOf(parts[5]);
linecount++;
}//endloop
double[][] arr2D = new double[][] {position, year, month, day, close};
INDArray x = Nd4j.createFromArray(arr2D);
return x;
}
我正在尝试复制 csvplotter 示例并使用单个输入/输出网络执行线性回归。
我如何将数组行(0)作为特征和数组行(4)作为标签加载?
更多信息:
System.out.println(ds.rank());
long[] l = ds.shape();
System.out.println(l[0] + " , " + l[1] + " - " + l.length);
System.out.println(ds.length());
结果是:
2,
5, 1260 -2
6300
为了清楚起见,这是我的问题:
for (int i = 0; i < nEpochs; i++) {
net.fit(d);
}
根据我尝试添加数据的方式导致各种错误
解决方案
虽然我没有得到答案,但我意识到了我的问题。根据 csv 绘图仪示例中的注释,我假设 indarray 的行被传递到输入。但是,实际传递给输入的是列。
通过转置 INDArray 并添加两列,我需要网络处理数据。
INDArray ds;
ds = ds.transpose();
DataSet ddd = new DataSet();
ddd.setFeatures(ds.getColumn(0, true)); //true maintains matrix instead of vector
ddd.setLabels(ds.getColumn(4, true));
ddd.dataSetBatches(500);
System.out.println(ddd);
我的打印输出:
===========INPUT===================
[[0],
[1.0000],
[2.0000],
...,
[1257.0000],
[1258.0000],
[1259.0000]]
=================OUTPUT==================
[[540.3100],
[536.7000],
[533.3300],
...,
[1431.8199],
[1439.2200],
[1436.3800]]
尽管培训不成功,但这确实回答了我最初的问题。
推荐阅读
- for-loop - 在 MQL4 中创建一个计算柱数的 for 循环
- android - 如何从 ArrayList 中获取所有值
(我只得到最后一个值)? - awk - 如何在awk中多次打印
- tensorflow - 在 TensorFlow BYOL 架构中加入指数移动平均线
- reactjs - 使用 ReactJS 在 Chrome 扩展项目中保存数据
- solidity - 松露合约部署失败,发件人无效
- python - Flask-Dance 不能在缓存数据库中放置超过一行
- php - Bash 测试命令 [[ ]] 在基于 Debian 的发行版上的 PHP 函数 shell_exec() 中不起作用
- mysql - 使用 whereRaw 过滤的最新原始数据上的子查询左连接计数 - Laravel
- python - 如何更优雅地在蛇游戏(pygame)中添加食物?