INDArrays 列表如何存储在文件中

Question

我正在做一个强化学习项目，我有一个 List<INDArray>，其中包含世界状态列表，第二个 List<INDArray> 包含动作预测和奖励值，索引对应于当前状态第一个List

我想将这些数据存储在硬盘上以供以后训练，我该如何实现？

以萨克斯为例，我们有：

List<INDArray> stateList = new ArrayList<>();
stateList.add(Nd4j.valueArrayOf(new int[]{3,3,3}, 5));
stateList.add(Nd4j.valueArrayOf(new int[]{3,3,3}, 6));

List<INDArray> valueList = new ArrayList<>();
valueList.add(Nd4j.create(new float[]{1, 2}));
valueList.add(Nd4j.create(new float[]{3, 4}));

Answer 1

您必须准备文件内容，然后简单地写入文件。

        String fileContent = "";
        for (INDArray arr : valueList) {
            str +=arr.getValue()+"/n";//arr.getValue() anything which u want to add
        }

        FileWriter fileWriter = new FileWriter("c:/temp/samplefile.txt");
        fileWriter.write(fileContent);
        fileWriter.close();