tf_agents.bandits.agents.neural_linucb_agent.NeuralLinUCBVariableCollection

A collection of variables used by NeuralLinUCBAgent.

tf_agents.bandits.agents.neural_linucb_agent.NeuralLinUCBVariableCollection(
    num_actions: int,
    encoding_dim: int,
    dtype: tf.DType = tf.float64,
    name: Optional[Text] = None
)

num_actions (int) number of actions the agent acts on.
encoding_dim (int) The dimensionality of the output of the encoding network.
dtype The type of the variables. Should be one of tf.float32 and tf.float64.
name (string) the name of this instance.