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 | ±----------±----------±-------±------------+