mirror of
https://github.com/wassname/ray.git
synced 2026-07-05 05:39:30 +08:00
eugenevinitsky
37076a9ff8
* working multi action distribution and multiagent model * currently working but the splits arent done in the right place * added shared models * added categorical support and mountain car example * now compatible with generalized advantage estimation * working multiagent code with discrete and continuous example * moved reshaper to utils * code review changes made, ppo action placeholder moved to model catalog, all multiagent code moved out of fcnet * added examples in * added PEP8 compliance * examples are mostly pep8 compliant * removed all flake errors * added examples to jenkins tests * fixed custom options bug * added lines to let docker file find multiagent tests * shortened example run length * corrected nits * fixed flake errors
15 lines
723 B
Python
15 lines
723 B
Python
from ray.rllib.models.catalog import ModelCatalog
|
|
from ray.rllib.models.action_dist import (ActionDistribution, Categorical,
|
|
DiagGaussian, Deterministic)
|
|
from ray.rllib.models.model import Model
|
|
from ray.rllib.models.fcnet import FullyConnectedNetwork
|
|
from ray.rllib.models.convnet import ConvolutionalNetwork
|
|
from ray.rllib.models.lstm import LSTM
|
|
from ray.rllib.models.multiagentfcnet import MultiAgentFullyConnectedNetwork
|
|
|
|
|
|
__all__ = ["ActionDistribution", "ActionDistribution", "Categorical",
|
|
"DiagGaussian", "Deterministic", "ModelCatalog", "Model",
|
|
"FullyConnectedNetwork", "ConvolutionalNetwork", "LSTM",
|
|
"MultiAgentFullyConnectedNetwork"]
|