Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

include energy kernel for MMD loss #262

Closed

Conversation

florence-bockting
Copy link

Situation/Background:
For the MMD loss only the Gaussian kernel and the inverse multiquadratic kernel are implemented
Changes made:
I implemented additionally the energy kernel as suggested in [1,2]
Still ToDo:
Sofar the kernel for the MMD loss can't be selected by the user but is by default the "gaussian" kernel.

[1] Feydy, J., Séjourné, T., Vialard, F. X., Amari, S. I., Trouvé, A., & Peyré, G. (2019, April). Interpolating between optimal transport and mmd using sinkhorn divergences. In The 22nd International Conference on Artificial Intelligence and Statistics (pp. 2681-2690). PMLR.
[2] Feydy, J. Geometric Data Analysis, beyond Convolutions. Ph.D. Thesis, Université Paris-Saclay, Gif-sur-Yvette, France, 2020.

@paul-buerkner
Copy link
Contributor

Thank you for this PR! As discussed offline, I will close the PR as it was made against master rather than dev.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants