28 lines
539 B
Python
28 lines
539 B
Python
"""
|
|
MAGAIL Algorithm Package
|
|
|
|
多智能体生成对抗模仿学习算法实现
|
|
"""
|
|
|
|
from .magail import MAGAIL
|
|
from .ppo import PPO
|
|
from .disc import GAILDiscrim
|
|
from .bert import Bert
|
|
from .policy import StateIndependentPolicy
|
|
from .buffer import RolloutBuffer
|
|
from .utils import Normalizer, build_mlp, reparameterize, evaluate_lop_pi
|
|
|
|
__all__ = [
|
|
'MAGAIL',
|
|
'PPO',
|
|
'GAILDiscrim',
|
|
'Bert',
|
|
'StateIndependentPolicy',
|
|
'RolloutBuffer',
|
|
'Normalizer',
|
|
'build_mlp',
|
|
'reparameterize',
|
|
'evaluate_lop_pi',
|
|
]
|
|
|