mirror of
https://github.com/facebookresearch/ReAgent.git
synced 2026-06-16 12:44:41 +00:00
Badri Narayan Bhaskar
0f3de8417b
Reviewed By: kaiwenw Differential Revision: D24765530 fbshipit-source-id: 196053173d748f380476647f3ac1219d40e1ea66
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
|
|
|
|
from .actor import (
|
|
DirichletFullyConnectedActor,
|
|
FullyConnectedActor,
|
|
GaussianFullyConnectedActor,
|
|
)
|
|
from .base import ModelBase
|
|
from .bcq import BatchConstrainedDQN
|
|
from .categorical_dqn import CategoricalDQN
|
|
from .containers import Sequential
|
|
from .critic import FullyConnectedCritic
|
|
from .dqn import FullyConnectedDQN
|
|
from .dueling_q_network import DuelingQNetwork, ParametricDuelingQNetwork
|
|
from .embedding_bag_concat import EmbeddingBagConcat
|
|
from .fully_connected_network import FullyConnectedNetwork
|
|
from .mlp_scorer import MLPScorer
|
|
from .seq2reward_model import Seq2RewardNetwork
|
|
|
|
|
|
__all__ = [
|
|
"ModelBase",
|
|
"Sequential",
|
|
"FullyConnectedDQN",
|
|
"DuelingQNetwork",
|
|
"ParametricDuelingQNetwork",
|
|
"BatchConstrainedDQN",
|
|
"CategoricalDQN",
|
|
"EmbeddingBagConcat",
|
|
"FullyConnectedNetwork",
|
|
"FullyConnectedCritic",
|
|
"GaussianFullyConnectedActor",
|
|
"DirichletFullyConnectedActor",
|
|
"FullyConnectedActor",
|
|
"MLPScorer",
|
|
"Seq2RewardNetwork",
|
|
]
|