def get_argument(parser=None):
parser = OffPolicyAgent.get_argument(parser)
parser.add_argument('--enable-double-dqn', action='store_true')
parser.add_argument('--enable-dueling-dqn', action='store_true')
parser.add_argument('--enable-categorical-dqn', action='store_true')
parser.add_argument('--enable-noisy-dqn', action='store_true')
return parser
def setUpClass(cls):
cls.batch_size = 32
cls.memory_capacity = 32
cls.on_policy_agent = OnPolicyAgent(name="OnPolicyAgent",
batch_size=cls.batch_size)
cls.off_policy_agent = OffPolicyAgent(
name="OffPolicyAgent", memory_capacity=cls.memory_capacity)
cls.discrete_env = gym.make("CartPole-v0")
cls.continuous_env = gym.make("Pendulum-v0")
def get_argument(parser=None):
"""
Create or update argument parser for command line program
Args:
parser (argparse.ArgParser, optional): argument parser
Returns:
argparse.ArgParser: argument parser
"""
parser = OffPolicyAgent.get_argument(parser)
parser.add_argument('--alpha', type=float, default=0.2)
parser.add_argument('--auto-alpha', action="store_true")
return parser
def get_argument(parser=None):
"""
Create or update argument parser for command line program
Args:
parser (argparse.ArgParser, optional): argument parser
Returns:
argparse.ArgParser: argument parser
"""
parser = OffPolicyAgent.get_argument(parser)
parser.add_argument('--enable-double-dqn', action='store_true')
parser.add_argument('--enable-dueling-dqn', action='store_true')
parser.add_argument('--enable-noisy-dqn', action='store_true')
return parser
def get_argument(parser=None):
parser = OffPolicyAgent.get_argument(parser)
parser.add_argument('--alpha', type=float, default=0.2)
parser.add_argument('--auto-alpha', action="store_true")
return parser
