BeautyOfWeb / AffinityNet

AffinityNet with feature attention layer and kNN attention pooling layer for few-shot semi-supervised learning
MIT License
38 stars 5 forks source link

AffinityNet

AffinityNet with stacked kNN attention pooling layers for few-shot semi-supervised learning

This repository is associated with the paper: AffinityNet: semi-supervised few-shot learning for disease type prediction

For any questions about the code, please contact tianlema@buffalo.edu