YLKnight / DKN

3 stars 0 forks source link

Multi-class Classification #2

Open howardchanth opened 1 month ago

howardchanth commented 1 month ago

Hi thanks for the great work! I am wondering if it is possible to use DKN to perform multi-class classification (e.g., 10 classes of CIFAR 10). Any guide lines or hint would be must appreciated. I tried to use R = 10 (to represent the number of classes? please correct me if I am wrong) and still the logistics regression but I encountered the following error. I think the block shape list need to be adjusted correspondingly. Thanks a lot for your time and help!

9eb8b6bdfb27e34fd272c894717cef1