alanqrwang/nwhead
Code for the Nadaraya-Watson Head - an interpretable/explainable, nonparametric classification head which can be used with any neural network
Python
Code for the Nadaraya-Watson Head - an interpretable/explainable, nonparametric classification head which can be used with any neural network
Python