Skin Cancer MNIST: HAM10000 (https://www.kaggle.com/kmader/skin-cancer-mnist-ham10000/home)
Resnet50
Model | Prec@1 |
---|---|
Resnet50 | 0.8922155689 |
ResNet50+Attention | 0.8942115768 |
ResNet50+Weighted Cross Entropy | 0.8882235529 |
sh ./train.sh
- Pytorch0.4.1
- python3.6.7
Data imbalance
微软面试 offline test。主要是想解决数据不均衡问题(Data Augmentation + Weighted Cross Entropy)。