ray/rllib/models
Matthew A. Wright e3c9f7e83a Custom action distributions (#5164)
* custom action dist wip

* Test case for custom action dist

* ActionDistribution.get_parameter_shape_for_action_space pattern

* Edit exception message to also suggest using a custom action distribution

* Clean up ModelCatalog.get_action_dist

* Pass model config to ActionDistribution constructors

* Update custom action distribution test case

* Name fix

* Autoformatter

* parameter shape static methods for torch distributions

* Fix docstring

* Generalize fake array for graph initialization

* Fix action dist constructors

* Correct parameter shape static methods for multicategorical and gaussian

* Make suggested changes to custom action dist's

* Correct instances of not passing model config to action dist

* Autoformatter

* fix tuple distribution constructor

* bugfix
2019-08-06 11:13:16 -07:00
..
tf Custom action distributions (#5164) 2019-08-06 11:13:16 -07:00
torch Custom action distributions (#5164) 2019-08-06 11:13:16 -07:00
__init__.py [rllib] Try moving RLlib to top level dir (#5324) 2019-08-05 23:25:49 -07:00
action_dist.py Custom action distributions (#5164) 2019-08-06 11:13:16 -07:00
catalog.py Custom action distributions (#5164) 2019-08-06 11:13:16 -07:00
extra_spaces.py [rllib] Try moving RLlib to top level dir (#5324) 2019-08-05 23:25:49 -07:00
model.py [rllib] Try moving RLlib to top level dir (#5324) 2019-08-05 23:25:49 -07:00
modelv2.py [rllib] Try moving RLlib to top level dir (#5324) 2019-08-05 23:25:49 -07:00
preprocessors.py [rllib] Try moving RLlib to top level dir (#5324) 2019-08-05 23:25:49 -07:00
README.txt [rllib] Try moving RLlib to top level dir (#5324) 2019-08-05 23:25:49 -07:00

Shared neural network models for RLlib.