org.deeplearning4j.rl4j.space

Class DiscreteSpace

java.lang.Object

org.deeplearning4j.rl4j.space.DiscreteSpace

All Implemented Interfaces:

ActionSpace<Integer>

Direct Known Subclasses:

HighLowDiscrete

public class DiscreteSpace
extends Object
implements ActionSpace<Integer>

Author:

rubenfiszel (ruben.fiszel@epfl.ch) on 7/8/16.

A discrete space of action. A discrete space is always isomorphic to a space of integer so we can parametrize directly by Integer. Benefit of using Integers directly is that you can use it as the id of the node assigned to that action in the outpout of a DQN.

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.api.rng.Random	rnd
protected int	size

Constructor Summary

Constructors

Constructor and Description
DiscreteSpace(int size)
DiscreteSpace(int size, org.nd4j.linalg.api.rng.Random rnd)

Method Summary

Modifier and Type	Method and Description
Object	encode(Integer a)
Integer	noOp()
Integer	randomAction()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.deeplearning4j.rl4j.space.ActionSpace

getSize

Field Detail

size

protected final int size

rnd

protected final org.nd4j.linalg.api.rng.Random rnd

Constructor Detail

DiscreteSpace

public DiscreteSpace(int size)

DiscreteSpace

public DiscreteSpace(int size,
                     org.nd4j.linalg.api.rng.Random rnd)

Method Detail

randomAction

public Integer randomAction()

Specified by:

randomAction in interface ActionSpace<Integer>

Returns:

A random action,

encode

public Object encode(Integer a)

Specified by:

encode in interface ActionSpace<Integer>

noOp

public Integer noOp()

Specified by:

noOp in interface ActionSpace<Integer>