@Test public void when_callingCreateFeatureArray_expect_INDArrayWithCorrectShape() { long[] observationShape = new long[] { 1, 2, 1 }; INDArray result = sut.createFeatureArray(4, observationShape); assertArrayEquals(new long[] { 1, 2, 4 }, result.shape()); }
@Override protected INDArray createFeatureArray(int size, long[] observationShape) { return INDArrayHelper.createRnnBatchForShape(size, observationShape); }
RecurrentActorCriticHelper extends ActorCriticHelper { @Override protected INDArray createFeatureArray(int size, long[] observationShape) { return INDArrayHelper.createRnnBatchForShape(size, observationShape); } }
RecurrentActorCriticHelper extends ActorCriticHelper { @Override protected INDArray createFeatureArray(int size, long[] observationShape) { return INDArrayHelper.createRnnBatchForShape(size, observationShape); } RecurrentActorCriticHelper(int actionSpaceSize); }
RecurrentActorCriticHelper extends ActorCriticHelper { @Override protected INDArray createFeatureArray(int size, long[] observationShape) { return INDArrayHelper.createRnnBatchForShape(size, observationShape); } 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 protected INDArray createFeatureArray(int size, long[] observationShape) { return INDArrayHelper.createRnnBatchForShape(size, observationShape); } RecurrentActorCriticHelper(int actionSpaceSize); @Override INDArray createValueLabels(int trainingBatchSize); @Override INDArray createPolicyLabels(int trainingBatchSize); @Override void setPolicy(INDArray policy, long idx, int action, double advantage);  }