如何将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();