Hello! Just curious to know if this model works for Binary class classification prob. I had tried it but seems no luck!

BayesWatch / deep-kernel-transfer

Official pytorch implementation of the paper "Bayesian Meta-Learning for the Few-Shot Setting via Deep Kernels" (NeurIPS 2020)

https://arxiv.org/abs/1910.05199

197 stars 29 forks source link

Hello! Just curious to know if this model works for Binary class classification prob. I had tried it but seems no luck! #19

Closed Shantan243 closed 1 year ago

mpatacchiola commented 1 year ago

Hi @Shantan243, yes the method can be used to deal with binary classification problems. If your dataset is different from those used in our work then you may need to tune the hyper-parameters and training schedule to get best results.

Shantan243 commented 1 year ago

Thanks @mpatacchiola. It has worked for me..