search

moskomule / anatome

Ἀνατομή is a PyTorch library to analyze representation of neural networks
MIT License
61 stars 6 forks source link

Correcting CKA similarity to output distance by subtracting 1 #7

Closed brando90 closed 3 years ago

brando90 commented 3 years ago

corrected what seems to be that anatome is returning similarity when the function name is distance.

Including the table from the original CKA paper to ensure my suggestion is correct indeed:

Screen Shot 2021-02-02 at 4 52 02 PM

paper link: http://proceedings.mlr.press/v97/kornblith19a.html

brando90 commented 3 years ago

original issue: https://github.com/moskomule/anatome/issues/6

brando90 commented 3 years ago

let me know if I may be of any further help. If you think I am incorrect please feel free to let me know.

moskomule commented 3 years ago

Thank you for your contribution! That should fix the problem.