breast_classification_ResNet 训练结果
运用了迁移学习,下载ResNet网络的权重文件,不改变原网络结构,在最后加上一个全局平均池化层、1024的FC、2的FC
实验一:
batch_size = 16
epochs = 100
初始学习率:0.0002
训练结果
实验一:
batch_size = 16
epochs = 100
初始学习率:0.0002
训练结果
混淆矩阵:
the model accuracy is 0.671875
±----------±----------±-------±------------+
| | Precision | Recall | Specificity |
±----------±----------±-------±------------+
| benign | 0.672 | 1.0 | 0.0 |
| malignant | nan | 0.0 | 1.0 |
±----------±----------±-------±------------+