ray.rllib.core.models.catalog.Catalog.get_action_dist_cls#

Catalog.get_action_dist_cls(framework: str)[源代码]#

获取动作分布类。

默认行为是从 Catalog._action_dist_class_fn 获取动作分布。

你应该重写这个方法,以便 RLlib 构建你的自定义动作分布,而不是默认的分布。例如,如果你不想使用 RLlib 的默认 RLModules 及其默认模型,而只想更改 Catalog 返回的分布。

参数:

framework – 要使用的框架。可以是“torch”或“tf2”。

返回:

动作分布。