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

extend comment on divergence

Browse Source
This commit is contained in:
Johannes Fischer
2021-08-02 19:15:02 +02:00
parent f468b3b7a4
commit b869597717
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/metrics.py
View File

@@ -83,6 +83,11 @@ def divergence(p, q, type='kl', n_components=0):
Args: Args:
p (torch.tensor): (n) samples from p p (torch.tensor): (n) samples from p
q (torch.tensor): (m) samples from q q (torch.tensor): (m) samples from q
type (str): divergence to use
n_components (int): method to use to compute kl divergence
n_components < 0: approximate samples with histogram density
n_components == 0: approximate samples by Gaussian distributions and compute analytically
n_components > 0: approximate samples as Gaussian mixture models with n_components components
Returns: Returns:
d (float): approximate divergence d (float): approximate divergence
""" """