ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf_agents.bandits.agents.utils

Common utility code and linear algebra functions.


build_laplacian_nearest_neighbor_graph(...): Build the Laplacian matrix of a nearest neighbor graph.

build_laplacian_over_ordinal_integer_actions(...): Build the unnormalized Laplacian matrix over ordinal integer actions.

compute_pairwise_distances(...): Compute the pairwise distances matrix.

process_experience_for_neural_agents(...): Processes the experience and prepares it for the network of the agent.

sum_reward_weighted_observations(...): Calculates an update used by some Bandit algorithms.