prasuchit/mmap-birl

Marginal MAP Bayesian Inverse Reinforcement Learning for dealing with occlusions and noise in learning data.

mmap-birl's Issues

IndexError in generator.py

Please consider that your code has the index error message while running your code as it is with no changes:

Traceback (most recent call last):
File "/mmap-testing/mmap-birl-master/code/runner.py", line 126, in
main()
File "/mmap-testing/mmap-birl-master/code/runner.py", line 51, in main
expertData = generator.generateDemonstration(mdp, problem)
File "/mmap-testing/mmap-birl-master/code/generator.py", line 56, in generateDemonstration
trajs, policy = generateTrajectory(mdp, problem)
File "/mmap-testing/mmap-birl-master/code/generator.py", line 124, in generateTrajectory
trajs, trajVmean, trajVvar = sampleTrajectories(problem.nTrajs, problem.nSteps, policy, mdp, problem.seed)
File "/mmap-testing/mmap-birl-master/code/generator.py", line 232, in sampleTrajectories
a = np.squeeze(piL[s])
IndexError: index 140 is out of bounds for axis 0 with size 48

Solving it requires uncommenting line 227 and commenting line 229 in your generator.py code.

Please consider if that is correct or please push a latest working code of mmap-birl on your github repo.