org.deeplearning4j.rl4j.mdp.gym.ActionTransformer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rl4j-gym Show documentation

There is a newer version: 1.0.0-beta

package org.deeplearning4j.rl4j.mdp.gym;


import org.deeplearning4j.rl4j.space.DiscreteSpace;
import org.deeplearning4j.rl4j.space.HighLowDiscrete;

/**
 * @author rubenfiszel ([email protected]) on 7/28/16.
 *
 * this can be used to convert the discrete domain (1 action by step)
 * into the HighLow from Gym
 */

public class ActionTransformer extends DiscreteSpace {

    final private int[] availableAction;
    final private HighLowDiscrete hld;

    public ActionTransformer(HighLowDiscrete hld, int[] availableAction) {
        super(availableAction.length);
        this.hld = hld;
        this.availableAction = availableAction;
    }

    @Override
    public Object encode(Integer a) {
        return hld.encode(availableAction[a]);
    }
}