forked from lassepe/GameInference.jl
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSimulationTest.jl
25 lines (21 loc) · 933 Bytes
/
SimulationTest.jl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
np = 3
rng = MersenneTwister(1)
tempDir = joinpath(@__DIR__, "../results/test")
problemTest = setup_problem(np)
# Prediction experiment
# ego_basetype = MAPController
# ego_controller = ego_basetype{nothing}()
# eval_controller = FixedSeedController{@S(1:n_controls(nominal_game(problemTest.d_eval)))}()
# Planning experiment
uids = uindex(nominal_game(problemTest.d_eval))
uid_ego = first(uids)
uid_eval = vcat(uids[2:end]...)
# ego_controller = MAPController{uid_ego}()
# eval_controller = MAPController{uid_eval}()
ego_controller = MAPController{@S(1:n_controls(nominal_game(problemTest.d_eval)))}()
eval_controller = MAPController{nothing}()
traj_cl, ego_prediction, true_game = run_simulation(problemTest..., eval_controller, ego_controller, rng; n_particles=60,
infer_only=true, visualize=true, save_fig_path=nothing)
global out_traj_cl = traj_cl
global out_ego_prediction = ego_prediction
global out_true_game = true_game