* 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>
19 lines
339 B
Java
19 lines
339 B
Java
package org.deeplearning4j.rl4j.support;
|
|
|
|
import org.deeplearning4j.rl4j.space.Encodable;
|
|
|
|
public class MockEncodable implements Encodable {
|
|
|
|
private final int value;
|
|
|
|
public MockEncodable(int value) {
|
|
|
|
this.value = value;
|
|
}
|
|
|
|
@Override
|
|
public double[] toArray() {
|
|
return new double[] { value };
|
|
}
|
|
}
|