Module: tf_agents.bandits.networks.global_and_arm_feature_network

View source on GitHub

Networks that take as input global and per-arm features, and output rewards.

Classes

class GlobalAndArmCommonTowerNetwork: A network that takes global and arm observations and outputs rewards.

class GlobalAndArmDotProductNetwork: A network that takes global and arm observations and outputs rewards.

Functions

create_feed_forward_common_tower_network(...): Creates a common tower network with feedforward towers.

create_feed_forward_dot_product_network(...): Creates a dot product network with feedforward towers.