* Added a unit test to help refac QLearningDiscrete.trainStep() Signed-off-by: unknown <aboulang2002@yahoo.com> * Changed expReplay setter to package private Signed-off-by: Alexandre Boulanger <aboulang2002@yahoo.com>
28 lines
511 B
Java
28 lines
511 B
Java
package org.deeplearning4j.rl4j.support;
|
|
|
|
import org.deeplearning4j.rl4j.space.ObservationSpace;
|
|
import org.nd4j.linalg.api.ndarray.INDArray;
|
|
|
|
public class MockObservationSpace implements ObservationSpace {
|
|
|
|
@Override
|
|
public String getName() {
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public int[] getShape() {
|
|
return new int[] { 1 };
|
|
}
|
|
|
|
@Override
|
|
public INDArray getLow() {
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public INDArray getHigh() {
|
|
return null;
|
|
}
|
|
}
|