facebookresearch / odin

A simple and effective method for detecting out-of-distribution images in neural networks.
Other
526 stars 102 forks source link

Do you provide code somewhere for calculating the Maximum Mean Discrepancy? #16

Open praveen5733 opened 3 years ago

praveen5733 commented 3 years ago

There are comparisons in the paper for OOD detection performance VS MMD of the datasets. Have you provided the code for calculating the MMD somewhere or is it possible to provide?

YixuanLi commented 3 years ago

Hi Praveen,

Here is a more well-documented library that might be useful: https://torch-two-sample.readthedocs.io/en/latest/. Hope it helps.