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

Use JS divergence in metrics

Browse Source
This commit is contained in:
Johannes Fischer
2021-08-03 17:04:47 +02:00
parent 1ee46214a7
commit 5919a4e439
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/metrics.py
View File

@@ -53,15 +53,15 @@ def metrics(filestr: str, test_dataset, policy):
visualize_distribution(true_actions[:,0], pred_actions[:,0], filestr+'_action_viz') visualize_distribution(true_actions[:,0], pred_actions[:,0], filestr+'_action_viz')
# calculate divergence between acceleration distributions # calculate divergence between acceleration distributions
acceleration_kl = divergence(pred_actions, true_actions, type='kl', n_components=-1) acceleration_divergence = divergence(pred_actions, true_actions, type='js', n_components=-1)
info['acceleration_kl'] = acceleration_kl info['acceleration_divergence'] = acceleration_divergence
# calculate divergence between velocity distributions # calculate divergence between velocity distributions
sim_velocities = states[:,:,2] sim_velocities = states[:,:,2]
sim_velocities = sim_velocities[~torch.isnan(sim_velocities)].flatten() sim_velocities = sim_velocities[~torch.isnan(sim_velocities)].flatten()
true_velocities = torch.cat(true_velocities, dim=0) true_velocities = torch.cat(true_velocities, dim=0)
velocity_kl = divergence(sim_velocities, true_velocities, type='kl', n_components=-1) velocity_divergence = divergence(sim_velocities, true_velocities, type='js', n_components=-1)
info['velocity_kl'] = velocity_kl info['velocity_divergence'] = velocity_divergence
return info return info