12345678910111213141516171819202122232425262728293031323334353637 |
- """Registry of algorithm names for `rllib train --run=<alg_name>`"""
- def _import_random_agent():
- from ray.rllib.contrib.random_agent.random_agent import RandomAgent
- return RandomAgent, RandomAgent._default_config
- def _import_maddpg():
- from ray.rllib.contrib import maddpg
- return maddpg.MADDPGTrainer, maddpg.DEFAULT_CONFIG
- def _import_alphazero():
- from ray.rllib.contrib.alpha_zero.core.alpha_zero_trainer import\
- AlphaZeroTrainer, DEFAULT_CONFIG
- return AlphaZeroTrainer, DEFAULT_CONFIG
- def _import_bandit_lints():
- from ray.rllib.contrib.bandits.agents.lin_ts import LinTSTrainer, TS_CONFIG
- return LinTSTrainer, TS_CONFIG
- def _import_bandit_linucb():
- from ray.rllib.contrib.bandits.agents.lin_ucb import LinUCBTrainer, \
- UCB_CONFIG
- return LinUCBTrainer, UCB_CONFIG
- CONTRIBUTED_ALGORITHMS = {
- "contrib/RandomAgent": _import_random_agent,
- "contrib/MADDPG": _import_maddpg,
- "contrib/AlphaZero": _import_alphazero,
- "contrib/LinTS": _import_bandit_lints,
- "contrib/LinUCB": _import_bandit_linucb
- }
|