INNER CODE UNIT · Python
build_dict_from_trajectory
GoogleCloudPlatform/vertex-ai-samples · community-content/tf_agents_bandits_movie_recommendation_with_kfp_and_vertex_sdk/mlops_pipeline_tf_agents_bandits_movie_recommendation/src/logger/main.py:143
def build_dict_from_trajectory(
trajectory: trajectories.Trajectory) -> Dict[str, Any]:
"""Builds a dict from `trajectory` data.
Args:
trajectory: A `trajectories.Trajectory` object.
Returns:
A dict holding the same data as `trajectory`.
"""
trajectory_dict = {
"step_type": trajectory.step_type.numpy().tolist(),
"observation": [{
"observation_batch": batch
} for batch in trajectory.observation.numpy().tolist()],
"action": trajectory.action.numpy().tolist(),
"policy_info": trajectory.policy_info,
"next_step_type": trajectory.next_step_type.numpy().tolist(),