Finish up load expert trajectories - Githubissues

GAIL-4-BARK / bark-ml

Machine Learning Applied to Autonomous Driving

MIT License

0 stars 0 forks source link

Finish up load expert trajectories #42

Closed MarcelBruckner closed 4 years ago

MarcelBruckner commented 4 years ago

Please look into bark_ml/library_wrappers/lib_tf2rl/load_expert_trajectories.py. I implemented the load for all .pkl files and the flattening like in https://github.com/keiohta/tf2rl/blob/master/tf2rl/experiments/utils.py.

[x] Load all pickle files in folder
[x] Flatten all observations and actions of all files and agents into a single huge dict
[x] Calculate the dt