ray/rllib/algorithms/dqn/__init__.py

11 lines
310 B
Python

from ray.rllib.algorithms.dqn.dqn import DQN, DQNConfig, DEFAULT_CONFIG
from ray.rllib.algorithms.dqn.dqn_tf_policy import DQNTFPolicy
from ray.rllib.algorithms.dqn.dqn_torch_policy import DQNTorchPolicy
__all__ = [
"DQN",
"DQNConfig",
"DQNTFPolicy",
"DQNTorchPolicy",
"DEFAULT_CONFIG",
]