When I used the parameters you provided and the pre-trained model, I used the DDSM data set to fine-tune the network, but the AUC value was 0.5. I wonder why the results in your paper could not be reached

nyukat / BIRADS_classifier

High-resolution breast cancer screening with multi-view deep convolutional neural networks

https://arxiv.org/abs/1703.07047

BSD 2-Clause "Simplified" License

148 stars 58 forks source link

When I used the parameters you provided and the pre-trained model, I used the DDSM data set to fine-tune the network, but the AUC value was 0.5. I wonder why the results in your paper could not be reached #10

Closed zoupei123456 closed 5 years ago

zoupei123456 commented 5 years ago

I'm using the loss function:cross_entropy = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(labels=y, logits=y_prediction_birads)),The optimizer is :train_step = tf.train.AdamOptimizer(lr).minimize(cross_entropy)