22 lines
642 B
Python
22 lines
642 B
Python
from tianshou.policy.base import BasePolicy
|
|
from tianshou.policy.imitation.base import ImitationPolicy
|
|
from tianshou.policy.modelfree.dqn import DQNPolicy
|
|
from tianshou.policy.modelfree.pg import PGPolicy
|
|
from tianshou.policy.modelfree.a2c import A2CPolicy
|
|
from tianshou.policy.modelfree.ddpg import DDPGPolicy
|
|
from tianshou.policy.modelfree.ppo import PPOPolicy
|
|
from tianshou.policy.modelfree.td3 import TD3Policy
|
|
from tianshou.policy.modelfree.sac import SACPolicy
|
|
|
|
__all__ = [
|
|
'BasePolicy',
|
|
'ImitationPolicy',
|
|
'DQNPolicy',
|
|
'PGPolicy',
|
|
'A2CPolicy',
|
|
'DDPGPolicy',
|
|
'PPOPolicy',
|
|
'TD3Policy',
|
|
'SACPolicy',
|
|
]
|