Я работаю над проектом обучения с подкреплением, и у меня есть 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}));