huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update scratch

Browse Source
This commit is contained in:
Johannes Fischer
2021-10-28 18:31:52 +02:00
parent 1a1f6d8836
commit 673b565e11
4 changed files with 24 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/evaluation/evaluation.py
View File

@@ -65,7 +65,7 @@ class Evaluation:
policy_accelerations = torch.tensor(self._accelerations)
metrics['acceleration_divergence'] = divergence(policy_accelerations, self.expert_accelerations, type='js')
visualize_distribution(self.expert_accelerations, policy_accelerations, os.path.join(self.filestr, '_action_viz{:02}'.format(epoch))
visualize_distribution(self.expert_accelerations, policy_accelerations, os.path.join(self.filestr, '_action_viz{:02}'.format(epoch)))
print(metrics)
return metrics