Как сохранить список INDArrays в файле - PullRequest
1 голос
/ 27 мая 2019

Я работаю над проектом обучения с подкреплением, и у меня есть List<INDArray>, который содержит список состояний мира, и второй List<INDArray>, который содержит значения для предсказания действий и вознаграждений с индексом, соответствующим состояниямfirst 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}));

1 Ответ

0 голосов
/ 27 мая 2019

Вы должны подготовить содержимое файла, а затем просто записать в файл.

        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();
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...