@Test public void when_callingCreateValueLabels_expect_INDArrayWithCorrectShape() { INDArray result = sut.createValueLabels(4); assertArrayEquals(new long[] { 1, 1, 4 }, result.shape()); }
@Override public INDArray createValueLabels(int trainingBatchSize) { return Nd4j.create(1, 1, trainingBatchSize); }
RecurrentActorCriticHelper extends ActorCriticHelper { @Override public INDArray createValueLabels(int trainingBatchSize) { return Nd4j.create(1, 1, trainingBatchSize); } }
RecurrentActorCriticHelper extends ActorCriticHelper { @Override public INDArray createValueLabels(int trainingBatchSize) { return Nd4j.create(1, 1, trainingBatchSize); } RecurrentActorCriticHelper(int actionSpaceSize); }
RecurrentActorCriticHelper extends ActorCriticHelper { @Override public INDArray createValueLabels(int trainingBatchSize) { return Nd4j.create(1, 1, trainingBatchSize); } RecurrentActorCriticHelper(int actionSpaceSize); @Override INDArray createValueLabels(int trainingBatchSize); @Override INDArray createPolicyLabels(int trainingBatchSize); @Override void setPolicy(INDArray policy, long idx, int action, double advantage); }
RecurrentActorCriticHelper extends ActorCriticHelper { @Override public INDArray createValueLabels(int trainingBatchSize) { return Nd4j.create(1, 1, trainingBatchSize); } RecurrentActorCriticHelper(int actionSpaceSize); @Override INDArray createValueLabels(int trainingBatchSize); @Override INDArray createPolicyLabels(int trainingBatchSize); @Override void setPolicy(INDArray policy, long idx, int action, double advantage);  }