Index

A

AccelTurnrate (class in envs.action) action() (algorithms.RL_Algorithm.optimizers.utils.replay_memory.Transition property) action_space() (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy property) (envs.base.AutoEnv property) (envs.make.Env property) (envs.multi_agent_env.MultiAgentAutoEnv property) action_space_spec() (envs.base.AutoEnv method) (envs.multi_agent_env.MultiAgentAutoEnv method) add() (algorithms.dataset.utils.KeyValueReplayMemory method) add_E2() (src.Vec.VecSE2.VecSE2 method) add_kwargs_to_reset() (in module envs.utils) AgentClass (class in src.Basic.Vehicle) algorithms (module) algorithms.AGen (module) algorithms.AGen.critic (module) algorithms.AGen.critic.base (module) algorithms.AGen.critic.model (module) algorithms.AGen.critic.utils (module) algorithms.AGen.my_gaussian_gru_policy (module) algorithms.AGen.rls (module) algorithms.AGen.validate_utils (module) algorithms.dataset (module) algorithms.dataset.CriticDataset (module) algorithms.dataset.utils (module) algorithms.distribution (module) algorithms.distribution.base (module) algorithms.distribution.diagonal_gaussian (module)

algorithms.distribution.recurrent_diagonal_gaussian (module) algorithms.policy (module) algorithms.policy.GaussianGRUPolicy (module) algorithms.policy.GaussianMLPBaseline (module) algorithms.policy.GRUCell (module) algorithms.policy.GRUNetwork (module) algorithms.policy.MLP (module) algorithms.RL_Algorithm (module) algorithms.RL_Algorithm.GAIL (module) algorithms.RL_Algorithm.GAIL.gail (module) algorithms.RL_Algorithm.optimizers (module) algorithms.RL_Algorithm.optimizers.trpo (module) algorithms.RL_Algorithm.optimizers.utils (module) algorithms.RL_Algorithm.optimizers.utils.math (module) algorithms.RL_Algorithm.optimizers.utils.replay_memory (module) algorithms.RL_Algorithm.optimizers.utils.tools (module) algorithms.RL_Algorithm.optimizers.utils.torch (module) algorithms.RL_Algorithm.optimizers.utils.zfilter (module) algorithms.RL_Algorithm.utils (module) algorithms.utils (module) append() (algorithms.RL_Algorithm.optimizers.utils.replay_memory.Memory method) arclength_1() (in module src.splines) arclength_2() (in module src.splines) are_collinear() (in module src.Vec.geom.geom) assets_dir() (in module algorithms.RL_Algorithm.optimizers.utils.tools) atan() (src.Vec.VecE2.VecE2 method) (src.Vec.VecSE2.VecSE2 method) AutoEnv (class in envs.base)

B

batch_to_path_rewards() (in module algorithms.AGen.critic.utils) batches() (algorithms.dataset.CriticDataset.CriticDataset method) (algorithms.dataset.CriticDataset.Dataset method) Block (class in algorithms.AGen.critic.model) build_baseline() (in module algorithms.utils) (in module envs.build_env) build_critic() (in module algorithms.utils) build_feature_extractor() (in module feature_extractor.utils) build_hierarchy() (in module envs.build_env)

build_ngsim_env() (in module algorithms.AGen.validate_utils) (in module algorithms.utils) (in module envs.build_env) build_policy() (in module algorithms.AGen.validate_utils) (in module algorithms.utils) (in module envs.build_env) build_reward_handler() (in module algorithms.utils) (in module envs.build_env) build_space() (in module envs.utils)

C

cal_agent_rmse() (in module envs.utils) cal_avg() (in module envs.utils) cal_lookahead() (in module envs.utils) cal_lookahead_rmse() (in module envs.utils) cal_m_stability() (in module envs.utils) cal_overall_rmse() (in module envs.utils) cal_step_rmse() (in module envs.utils) calc_curve_length_1() (in module src.splines) calc_curve_length_2() (in module src.splines) calc_curve_param_given_arclen() (in module src.splines) capacity() (src.Record.record.SceneRecord property) CAR (src.Basic.Vehicle.AgentClass attribute) car_df_index() (in module src.ngsim_trajdata) carid_set() (in module src.ngsim_trajdata) CarLidarFeatureExtractor (class in feature_extractor.CarLidarFeatureExtractor) carsinframe() (in module src.ngsim_trajdata) clamp() (in module src.curves.CurvePt) (in module src.Vec.VecSE2) clamp_euclidean() (in module src.Vec.VecSE2) clean_data() (in module preprocessing.clean_holo) CollisionCheckResult (class in feature_extractor.collision_detection) compute_dynamics_jacobian() (in module src.trajectory_smoothing) compute_flat_grad() (in module algorithms.RL_Algorithm.optimizers.utils.torch) compute_lengths() (in module envs.utils) compute_n_batches() (in module algorithms.dataset.utils)

compute_observation_jacobian() (in module src.trajectory_smoothing) conjugate_gradients() (in module algorithms.RL_Algorithm.optimizers.trpo) connect() (in module src.Roadway.roadway) convert() (in module src.trajdata) (in module src.Vec.VecSE2) (src.Vec.VecSE2.VecSE2 method) convert_2_float() (in module feature_extractor.interface) convert_curves_feet_to_meters() (in module src.roadway) convert_raw_ngsim_to_trajdatas() (in module src.trajdata) ConvexPolygon (class in feature_extractor.collision_detection) copy() (in module src.trajdata) copyto() (in module src.Record.frame) CoreFeatureExtractor (class in feature_extractor.CoreFeatureExtractor) CPAMemory (class in feature_extractor.collision_detection) create_index_file() (in module envs.utils) create_lane() (in module preprocessing.clean_holo) Critic (class in algorithms.AGen.critic.base) CriticDataset (class in algorithms.dataset.CriticDataset) critique() (algorithms.AGen.critic.base.Critic method) cross_product() (in module src.Vec.geom.geom) csv2txt() (in module preprocessing.clean_holo) CurveIndex (class in src.curves.CurvePt) curveindex_end() (in module src.curves.CurvePt) CurveProjection (class in src.curves.CurvePt) CurvePt (class in src.curves.CurvePt) cyclic_shift_left() (in module feature_extractor.collision_detection)

D

Dataset (class in algorithms.dataset.CriticDataset) delete_by_id() (src.Record.frame.Frame method) deleteat() (src.Record.frame.Frame method) deltaangle() (in module src.Vec.VecSE2) DiagonalGaussian (class in algorithms.distribution.diagonal_gaussian) dict_get() (in module envs.utils) dim() (algorithms.distribution.base.Distribution property) (algorithms.distribution.diagonal_gaussian.DiagonalGaussian property) dist() (in module src.Vec.VecE2) dist2() (in module src.Vec.VecE2) dist_info_keys() (algorithms.distribution.base.Distribution property)

dist_info_specs() (algorithms.distribution.base.Distribution property) (algorithms.distribution.diagonal_gaussian.DiagonalGaussian property) dist_info_sym() (algorithms.AGen.my_gaussian_gru_policy.myGaussianGRUPolicy method) (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy method) Distribution (class in algorithms.distribution.base) distribution() (algorithms.AGen.my_gaussian_gru_policy.myGaussianGRUPolicy property) (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy property) div() (in module src.curves.CurvePt) dot_product() (in module src.Vec.geom.geom) draw_obs_noise() (in module src.trajectory_smoothing) draw_proc_noise() (in module src.trajectory_smoothing)

E

EKF() (in module src.trajectory_smoothing) empty() (in module feature_extractor.collision_detection) (src.Record.frame.Frame method) (src.Record.record.SceneRecord method) ensure_pts_sorted_by_min_polar_angle() (in module feature_extractor.collision_detection) entropy() (algorithms.distribution.base.Distribution method) (algorithms.distribution.diagonal_gaussian.DiagonalGaussian method) Env (class in envs.make) envs (module) envs.action (module)

envs.base (module) envs.build_env (module) envs.hyperparams (module) envs.make (module) envs.multi_agent_env (module) envs.utils (module) extract_features() (in module preprocessing.extract_feature) extract_ngsim_features() (in module preprocessing.extract_feature) extract_normalizing_env() (in module algorithms.utils) extract_wrapped_env() (in module algorithms.utils)

F

feature_extractor (module) feature_extractor.CarLidarFeatureExtractor (module) feature_extractor.collision_detection (module) feature_extractor.CoreFeatureExtractor (module) feature_extractor.feature_extractor (module) feature_extractor.FeatureState (module) feature_extractor.ForeForeFeatureExtractor (module) feature_extractor.Get (module) feature_extractor.interface (module) feature_extractor.neighbor_feature (module) feature_extractor.TemporalFeatureExtractor (module) feature_extractor.utils (module) feature_extractor.WellBehavedFeatureExtractor (module) feature_info() (feature_extractor.CarLidarFeatureExtractor.CarLidarFeatureExtractor method) (feature_extractor.CoreFeatureExtractor.CoreFeatureExtractor method) (feature_extractor.feature_extractor.MultiFeatureExtractor method) (feature_extractor.ForeForeFeatureExtractor.ForeForeFeatureExtractor method) (feature_extractor.TemporalFeatureExtractor.TemporalFeatureExtractor method) (feature_extractor.WellBehavedFeatureExtractor.WellBehavedFeatureExtractor method) feature_names() (feature_extractor.CarLidarFeatureExtractor.CarLidarFeatureExtractor method) (feature_extractor.CoreFeatureExtractor.CoreFeatureExtractor method) (feature_extractor.feature_extractor.MultiFeatureExtractor method) (feature_extractor.ForeForeFeatureExtractor.ForeForeFeatureExtractor method) (feature_extractor.TemporalFeatureExtractor.TemporalFeatureExtractor method) (feature_extractor.WellBehavedFeatureExtractor.WellBehavedFeatureExtractor method)

FeatureValue (class in feature_extractor.FeatureState) fill_infos_cache() (in module envs.utils) filter_given_trajectory() (in module src.trajdata) filter_trajectory() (in module src.trajdata) FilterTrajectoryResult (class in src.trajdata) findfirst() (src.Record.frame.Frame method) first_carid() (in module src.ngsim_trajdata) fit() (algorithms.policy.GaussianMLPBaseline.GaussianMLP method) (algorithms.policy.GaussianMLPBaseline.GaussianMLPBaseline method) fit_cubic_spline() (in module src.splines) ForeForeFeatureExtractor (class in feature_extractor.ForeForeFeatureExtractor) forward() (algorithms.AGen.critic.model.Block method) (algorithms.AGen.critic.model.ObservationActionMLP method) (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy method) (algorithms.policy.GaussianMLPBaseline.GaussianMLP method) (algorithms.policy.GRUCell.GRUCell method) (algorithms.policy.GRUNetwork.GRUNetwork method) (algorithms.policy.MLP.MLP method) Frame (class in src.Record.frame) frame_inbounds() (in module src.Record.record) Frenet (class in src.Basic.Vehicle)

G

GAIL (class in algorithms.RL_Algorithm.GAIL.gail) GaussianGRUPolicy (class in algorithms.policy.GaussianGRUPolicy) GaussianMLP (class in algorithms.policy.GaussianMLPBaseline) GaussianMLPBaseline (class in algorithms.policy.GaussianMLPBaseline) get_Acc() (in module feature_extractor.interface) get_AccFs() (in module feature_extractor.interface) get_action() (algorithms.AGen.my_gaussian_gru_policy.myGaussianGRUPolicy method) (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy method) get_actions() (algorithms.AGen.my_gaussian_gru_policy.myGaussianGRUPolicy method) (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy method) get_actions_with_prev() (algorithms.AGen.my_gaussian_gru_policy.myGaussianGRUPolicy method) (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy method) get_AngularRateF() (in module feature_extractor.Get) get_AngularRateG() (in module feature_extractor.Get) get_by_id() (src.Roadway.roadway.Roadway method) get_by_ind_roadway() (src.Roadway.roadway.Lane method) get_by_roadindex() (src.Roadway.roadway.Roadway method) get_by_tag() (src.Roadway.roadway.Roadway method) get_center() (src.Basic.Vehicle.Vehicle property) get_closest_perpendicular_point_between_points() (in module src.Roadway.roadway) get_collision_time() (in module feature_extractor.collision_detection) get_control_noise_in_control_space() (in module src.trajectory_smoothing) get_corresponding_roadway() (in module src.trajdata) get_curve_list_by_index() (in module src.curves.CurvePt) get_curve_projection() (in module src.curves.CurvePt) get_def() (in module src.Record.record) get_edge() (in module feature_extractor.collision_detection) get_elapsed_time_3() (in module src.Record.record) get_features() (envs.base.AutoEnv method) (envs.multi_agent_env.MultiAgentAutoEnv method) get_fim() (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy method) get_first_collision() (in module feature_extractor.collision_detection) get_flat_grad_from() (in module algorithms.RL_Algorithm.optimizers.utils.torch) get_flat_params_from() (in module algorithms.RL_Algorithm.optimizers.utils.torch) get_footpoint() (src.Basic.Vehicle.Vehicle property) get_frame_range() (in module src.ngsim_trajdata) get_ground_truth() (in module algorithms.AGen.validate_utils) get_intersection_time() (in module src.Vec.geom.projectile) get_Inv_TTC() (in module feature_extractor.Get)

get_Is_Colliding() (in module feature_extractor.Get) get_itr_snapshot() (algorithms.RL_Algorithm.GAIL.gail.GAIL method) get_Jerk() (in module feature_extractor.interface) get_kl() (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy method) get_lane_width() (in module src.Basic.Vehicle) get_LaneCurvature() (in module feature_extractor.Get) get_lerp_time_1() (in module src.curves.CurvePt) get_lerp_time_2() (in module src.curves.CurvePt) get_lerp_time_unclamped_1() (in module src.curves.CurvePt) get_lerp_time_unclamped_2() (in module src.curves.CurvePt) get_lerp_time_unclamped_3() (in module src.curves.CurvePt) get_log_prob() (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy method) get_markerdist_left() (in module src.Basic.Vehicle) get_markerdist_right() (in module src.Basic.Vehicle) get_multiagent_ground_truth() (in module algorithms.AGen.validate_utils) get_neighbor_fore_along_lane_1() (in module feature_extractor.neighbor_feature) get_neighbor_fore_along_lane_2() (in module feature_extractor.neighbor_feature) get_neighbor_fore_along_lane_3() (in module feature_extractor.neighbor_feature) get_observation_noise_covariance() (in module src.trajectory_smoothing) get_polar_angle() (in module src.Vec.geom.line_segment) get_poly_edge() (in module feature_extractor.collision_detection) get_PosFyaw() (in module feature_extractor.Get) get_process_noise_covariance() (in module src.trajectory_smoothing) get_RoadEdgeDist_Left() (in module feature_extractor.Get) get_RoadEdgeDist_Right() (in module feature_extractor.Get) get_scene() (in module src.Record.record) get_segid() (in module src.roadway) get_signed_area() (in module feature_extractor.collision_detection) get_Speed() (in module feature_extractor.Get) get_targetpoint_delta() (src.Basic.Vehicle.Vehicle method) get_TimeGap() (in module feature_extractor.Get) get_transform_control_noise_to_state_space() (in module src.trajectory_smoothing) get_TurnRateF() (in module feature_extractor.Get) get_TurnRateG() (in module feature_extractor.Get) get_vehicle() (in module src.Record.record) get_VelFs() (in module feature_extractor.Get) get_VelFt() (in module feature_extractor.Get) GRUCell (class in algorithms.policy.GRUCell) GRUNetwork (class in algorithms.policy.GRUNetwork)

H

has_next() (in module src.Roadway.roadway) has_prev() (in module src.Roadway.roadway)

has_segment() (src.Roadway.roadway.Roadway method) hypot() (src.Vec.VecE2.VecE2 method)

I

in_poly() (in module feature_extractor.collision_detection) index_closest_to_point() (in module src.curves.CurvePt) index_ngsim_trajectory() (in module envs.utils) inertial2body() (in module src.Vec.geom.geom) init() (src.Record.frame.Frame method) (src.Record.record.SceneRecord method) init_env() (algorithms.RL_Algorithm.GAIL.gail.GAIL method) initialize() (algorithms.AGen.rls.rls method) insert() (src.Record.record.SceneRecord method) integrate() (in module src.roadway)

inverse_ttc_to_ttc() (in module feature_extractor.FeatureState) is_between_segments_hi() (in module src.Roadway.roadway) is_between_segments_lo() (in module src.Roadway.roadway) is_colliding_1() (in module feature_extractor.collision_detection) is_colliding_2() (in module feature_extractor.collision_detection) is_in_entrances() (in module src.Roadway.roadway) is_in_exits() (in module src.Roadway.roadway) is_potentially_colliding() (in module feature_extractor.collision_detection) iscarinframe() (in module src.ngsim_trajdata) isempty() (in module feature_extractor.collision_detection)

K

keep_vehicle_subset() (in module envs.utils) KeyValueReplayMemory (class in algorithms.dataset.utils) kl() (algorithms.distribution.base.Distribution method) (algorithms.distribution.diagonal_gaussian.DiagonalGaussian method)

kl_sym() (algorithms.distribution.base.Distribution method) (algorithms.distribution.diagonal_gaussian.DiagonalGaussian method)

L

Lane (class in src.Roadway.roadway) LaneBoundary (class in src.Roadway.roadway) LaneConnection (class in src.Roadway.roadway) LaneTag (class in src.Roadway.roadway) lerp() (in module src.curves.CurvePt) (in module src.Vec.VecE2) (in module src.Vec.VecSE2) lerp_angle() (in module src.Vec.VecSE2) LidarSensor (class in feature_extractor.CarLidarFeatureExtractor) likelihood_ratio_sym() (algorithms.distribution.base.Distribution method) (algorithms.distribution.diagonal_gaussian.DiagonalGaussian method) line_search() (in module algorithms.RL_Algorithm.optimizers.trpo) LineSegment (class in src.Vec.geom.line_segment) ListRecord (class in src.Record.record) load() (algorithms.RL_Algorithm.GAIL.gail.GAIL method) load_args() (in module envs.hyperparams) load_critic() (algorithms.RL_Algorithm.GAIL.gail.GAIL method)

load_data() (in module envs.utils) load_dataset() (in module algorithms.dataset.utils) load_ngsim_trajdata() (in module src.ngsim_trajdata) (in module src.trajdata) load_ngsim_trajdatas() (in module envs.utils) load_param() (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy method) load_params() (in module algorithms.utils) load_policy() (algorithms.RL_Algorithm.GAIL.gail.GAIL method) load_trajdata() (in module src.trajdata) load_validate_data() (in module algorithms.AGen.validate_utils) load_x_feature_names() (in module envs.utils) log_diagnostics() (algorithms.AGen.my_gaussian_gru_policy.myGaussianGRUPolicy method) (algorithms.RL_Algorithm.GAIL.gail.GAIL method) log_likelihood() (algorithms.distribution.base.Distribution method) (algorithms.distribution.diagonal_gaussian.DiagonalGaussian method) log_likelihood_sym() (algorithms.distribution.base.Distribution method) (algorithms.distribution.diagonal_gaussian.DiagonalGaussian method)

M

make_env() (in module envs.make) mask() (algorithms.RL_Algorithm.optimizers.utils.replay_memory.Transition property) max_n_objects() (in module envs.utils) maybe_mkdir() (in module algorithms.utils) (in module envs.utils) mean() (algorithms.RL_Algorithm.optimizers.utils.zfilter.RunningStat property) Memory (class in algorithms.RL_Algorithm.optimizers.utils.replay_memory) merge() (algorithms.RL_Algorithm.utils.RewardHandler method) minkowksi_sum() (in module feature_extractor.collision_detection)

minkowski_difference() (in module feature_extractor.collision_detection) mirror() (in module feature_extractor.collision_detection) MLP (class in algorithms.policy.MLP) mod2pi() (src.Vec.VecSE2.VecSE2 method) mod2pi2() (in module src.Roadway.utils) MOTORCYCLE (src.Basic.Vehicle.AgentClass attribute) MultiAgentAutoEnv (class in envs.multi_agent_env) MultiFeatureExtractor (class in feature_extractor.feature_extractor) MvNormal (class in src.MvNormal.MvNormal) myGaussianGRUPolicy (class in algorithms.AGen.my_gaussian_gru_policy)

N

n() (algorithms.RL_Algorithm.optimizers.utils.zfilter.RunningStat property) n_lanes_left() (in module src.Roadway.roadway) n_lanes_right() (in module src.Roadway.roadway) n_objects_in_frame() (src.Record.record.ListRecord method) nbeams() (feature_extractor.CarLidarFeatureExtractor.LidarSensor property) NeighborLongitudinalResult (class in feature_extractor.neighbor_feature) next_lane() (in module src.Roadway.roadway) next_lane_point() (in module src.Roadway.roadway) next_state() (algorithms.RL_Algorithm.optimizers.utils.replay_memory.Transition property) nframes() (src.Record.record.ListRecord property) NGSIMRoadway (class in src.roadway) NGSIMTrajdata (class in src.ngsim_trajdata) nids() (src.Record.record.ListRecord property)

norm() (in module src.Vec.VecE2) normal_entropy() (in module algorithms.RL_Algorithm.optimizers.utils.math) normal_log_density() (in module algorithms.RL_Algorithm.optimizers.utils.math) normaliza_euclidean() (in module src.Vec.VecSE2) normalize() (in module algorithms.AGen.validate_utils) (in module envs.utils) normalize_env_reset_with_kwargs() (in module envs.utils) normalize_range() (in module algorithms.AGen.validate_utils) (in module envs.utils) normsquared() (in module src.Vec.VecE2) nstates() (src.Record.record.ListRecord property) nth_carid() (in module src.ngsim_trajdata) num_envs() (envs.make.Env property) (envs.multi_agent_env.MultiAgentAutoEnv method)

O

obs_names() (envs.base.AutoEnv method) (envs.make.Env method) (envs.multi_agent_env.MultiAgentAutoEnv method) observation_space() (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy property) (envs.make.Env property) observation_space_spec() (envs.base.AutoEnv method) (envs.multi_agent_env.MultiAgentAutoEnv method) ObservationActionMLP (class in algorithms.AGen.critic.model)

observe() (in module feature_extractor.CarLidarFeatureExtractor) (in module src.trajectory_smoothing) obtain_samples() (algorithms.RL_Algorithm.GAIL.gail.GAIL method) ones() (in module algorithms.RL_Algorithm.GAIL.gail) (in module algorithms.RL_Algorithm.optimizers.trpo) (in module algorithms.RL_Algorithm.optimizers.utils) (in module algorithms.RL_Algorithm.optimizers.utils.torch) optimize_policy() (algorithms.RL_Algorithm.GAIL.gail.GAIL method)

P

pad_tensor() (in module algorithms.dataset.utils) parameters() (algorithms.policy.GaussianMLPBaseline.GaussianMLPBaseline method) parse_args() (in module envs.hyperparams) parse_lane_connection() (in module src.Roadway.roadway) partition_list() (in module envs.utils) pastframe_inbounds() (in module src.Record.record) PEDESTRIAN (src.Basic.Vehicle.AgentClass attribute) polar() (in module src.Vec.VecE2) predict() (algorithms.AGen.rls.rls method) (algorithms.policy.GaussianMLPBaseline.GaussianMLP method) (algorithms.policy.GaussianMLPBaseline.GaussianMLPBaseline method) preprocessing (module) preprocessing.clean_holo (module) preprocessing.extract_feature (module) prev_lane() (in module src.Roadway.roadway) prev_lane_point() (in module src.Roadway.roadway) print_error() (in module envs.utils) process_samples() (algorithms.RL_Algorithm.GAIL.gail.GAIL method)

proj() (in module src.curves.CurvePt) (in module src.Vec.VecE2) proj_() (in module src.Vec.VecE2) proj_1() (in module src.Roadway.roadway) proj_2() (in module src.Roadway.roadway) Projectile (class in src.Vec.geom.projectile) propagate() (in module envs.action) pull_features() (feature_extractor.CarLidarFeatureExtractor.CarLidarFeatureExtractor method) (feature_extractor.CoreFeatureExtractor.CoreFeatureExtractor method) (feature_extractor.feature_extractor.MultiFeatureExtractor method) (feature_extractor.ForeForeFeatureExtractor.ForeForeFeatureExtractor method) (feature_extractor.TemporalFeatureExtractor.TemporalFeatureExtractor method) (feature_extractor.WellBehavedFeatureExtractor.WellBehavedFeatureExtractor method) push() (algorithms.RL_Algorithm.optimizers.utils.replay_memory.Memory method) (algorithms.RL_Algorithm.optimizers.utils.zfilter.RunningStat method) (in module feature_extractor.collision_detection) (src.Record.frame.Frame method) push_back_records() (src.Record.record.SceneRecord method)

R

rand_sample() (src.MvNormal.MvNormal.MvNormal method) random_sample_from_set_without_replacement() (in module envs.utils) read_boundaries() (in module src.roadway) read_centerlines() (in module src.roadway) read_def() (in module src.Basic.Vehicle) read_frame() (in module src.Record.record) read_roadway() (in module src.roadway) (in module src.Roadway.roadway) read_state() (in module src.Basic.Vehicle) read_trajdata() (in module src.Record.record) RecordFrame (class in src.Record.record) RecordState (class in src.Record.record) recurrent() (algorithms.AGen.my_gaussian_gru_policy.myGaussianGRUPolicy property) (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy property) render() (envs.make.Env method) reset() (algorithms.AGen.my_gaussian_gru_policy.myGaussianGRUPolicy method) (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy method) (envs.base.AutoEnv method) (envs.make.Env method) (envs.multi_agent_env.MultiAgentAutoEnv method)

reward() (algorithms.RL_Algorithm.optimizers.utils.replay_memory.Transition property) RewardHandler (class in algorithms.RL_Algorithm.utils) rls (class in algorithms.AGen.rls) RoadIndex (class in src.Roadway.roadway) RoadProjection (class in src.Roadway.roadway) RoadSegment (class in src.Roadway.roadway) Roadway (class in src.Roadway.roadway) RoadwayInputParams (class in src.roadway) rot() (in module src.Vec.VecE2) (in module src.Vec.VecSE2) rot180() (src.Vec.VecE2.VecE2 method) (src.Vec.VecSE2.VecSE2 method) rotl90() (src.Vec.VecE2.VecE2 method) (src.Vec.VecSE2.VecSE2 method) rotr90() (src.Vec.VecE2.VecE2 method) (src.Vec.VecSE2.VecSE2 method) RunningStat (class in algorithms.RL_Algorithm.optimizers.utils.zfilter)

S

sample() (algorithms.dataset.utils.KeyValueReplayMemory method) (algorithms.distribution.diagonal_gaussian.DiagonalGaussian method) (algorithms.RL_Algorithm.optimizers.utils.replay_memory.Memory method) sample_multiple_trajdata_vehicle() (in module envs.utils) sample_spline_1() (in module src.splines) sample_spline_2() (in module src.splines) sample_spline_curvature_1() (in module src.splines) sample_spline_curvature_2() (in module src.splines) sample_spline_derivative() (in module src.splines) sample_spline_derivative2() (in module src.splines) sample_spline_derivative_of_curvature_1() (in module src.splines) sample_spline_derivative_of_curvature_2() (in module src.splines) sample_spline_speed_1() (in module src.splines) sample_spline_speed_2() (in module src.splines) sample_spline_theta_1() (in module src.splines) sample_spline_theta_2() (in module src.splines) sample_trajdata_vehicle() (in module envs.utils) save_error() (in module envs.utils) save_params() (in module algorithms.utils) scale_euclidean() (in module src.Vec.VecSE2) SceneRecord (class in src.Record.record) select_batch_idxs() (in module algorithms.dataset.CriticDataset) select_multiple_trajdata_vehicle() (in module envs.utils) set() (feature_extractor.collision_detection.ConvexPolygon method) (src.Basic.Vehicle.Frenet method) (src.Basic.Vehicle.VehicleState method) set_cuda() (algorithms.policy.GaussianMLPBaseline.GaussianMLPBaseline method) set_dual_feature() (in module feature_extractor.feature_extractor) set_feature() (in module feature_extractor.feature_extractor) set_feature_missing() (in module feature_extractor.feature_extractor) set_flat_params_to() (in module algorithms.RL_Algorithm.optimizers.utils.torch) set_up_experiment() (in module algorithms.utils) shape() (algorithms.RL_Algorithm.optimizers.utils.zfilter.RunningStat property) show() (src.curves.CurvePt.CurvePt method) (src.Vec.VecE2.VecE2 method) (src.Vec.VecSE2.VecSE2 method) shutdown_worker() (algorithms.RL_Algorithm.GAIL.gail.GAIL method) sign() (in module src.Roadway.roadway) (in module src.Vec.geom.geom)

SimulationResults (class in src.trajectory_smoothing) SpeedLimit (class in src.Roadway.roadway) src (module) src.Basic (module) src.Basic.Vehicle (module) src.const (module) src.curves (module) src.curves.CurvePt (module) src.MvNormal (module) src.MvNormal.MvNormal (module) src.ngsim_trajdata (module) src.Record (module) src.Record.frame (module) src.Record.record (module) src.Roadway (module) src.roadway (module) src.Roadway.roadway (module) src.Roadway.utils (module) src.splines (module) src.trajdata (module) src.trajectory_smoothing (module) src.Vec (module) src.Vec.geom (module) src.Vec.geom.geom (module) src.Vec.geom.line_segment (module) src.Vec.geom.projectile (module) src.Vec.VecE2 (module) src.Vec.VecSE2 (module) start_worker() (algorithms.RL_Algorithm.GAIL.gail.GAIL method) state() (algorithms.RL_Algorithm.optimizers.utils.replay_memory.Transition property) state_info_specs() (algorithms.AGen.my_gaussian_gru_policy.myGaussianGRUPolicy property) (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy property) std() (algorithms.RL_Algorithm.optimizers.utils.zfilter.RunningStat property) step() (envs.base.AutoEnv method) (envs.make.Env method) (envs.multi_agent_env.MultiAgentAutoEnv method) (in module src.trajectory_smoothing) str2bool() (in module envs.utils) symmetric_exponential_moving_average() (in module src.trajdata)

T

TemporalFeatureExtractor (class in feature_extractor.TemporalFeatureExtractor) tensor() (in module algorithms.RL_Algorithm.GAIL.gail) (in module algorithms.RL_Algorithm.optimizers.trpo) (in module algorithms.RL_Algorithm.optimizers.utils) (in module algorithms.RL_Algorithm.optimizers.utils.torch) to_device() (in module algorithms.RL_Algorithm.optimizers.utils.torch)

to_oriented_bounding_box_1() (in module feature_extractor.collision_detection) to_oriented_bounding_box_2() (in module feature_extractor.collision_detection) train() (algorithms.AGen.critic.base.Critic method) (algorithms.RL_Algorithm.GAIL.gail.GAIL method) Transition (class in algorithms.RL_Algorithm.optimizers.utils.replay_memory) trpo_step() (in module algorithms.RL_Algorithm.optimizers.trpo) TRUCK (src.Basic.Vehicle.AgentClass attribute)

U

update() (algorithms.AGen.rls.rls method) (src.Record.record.SceneRecord method)

V

var() (algorithms.RL_Algorithm.optimizers.utils.zfilter.RunningStat property) vec_env_executor() (envs.make.Env method) VecE2 (class in src.Vec.VecE2) VecSE2 (class in src.Vec.VecSE2) vectorized() (algorithms.AGen.my_gaussian_gru_policy.myGaussianGRUPolicy property) (algorithms.policy.GaussianGRUPolicy.GaussianGRUPolicy property) (envs.make.Env property) (envs.multi_agent_env.MultiAgentAutoEnv method)

Vehicle (class in src.Basic.Vehicle) VehicleDef (class in src.Basic.Vehicle) VehicleState (class in src.Basic.Vehicle) VehicleSystem (class in src.trajectory_smoothing)

W

WellBehavedFeatureExtractor (class in feature_extractor.WellBehavedFeatureExtractor) write() (src.Basic.Vehicle.VehicleDef method) (src.Basic.Vehicle.VehicleState method) (src.Record.record.ListRecord method) (src.Record.record.RecordFrame method) (src.Roadway.roadway.LaneConnection method) (src.Roadway.roadway.RoadIndex method) (src.Roadway.roadway.Roadway method)

write_features() (in module preprocessing.extract_feature) write_trajectories() (in module envs.utils)

Z

zeros() (in module algorithms.RL_Algorithm.GAIL.gail) (in module algorithms.RL_Algorithm.optimizers.trpo) (in module algorithms.RL_Algorithm.optimizers.utils) (in module algorithms.RL_Algorithm.optimizers.utils.torch)

ZFilter (class in algorithms.RL_Algorithm.optimizers.utils.zfilter)