Skip to content
This repository was archived by the owner on Sep 10, 2024. It is now read-only.

Commit af4b572

Browse files
authored
Update Fine Tuning Torchvision Models.md
1 parent 0dd6a6c commit af4b572

1 file changed

Lines changed: 10 additions & 3 deletions

File tree

Tutorials/Fine Tuning Torchvision Models.md

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
6# Fine Tuning Torchvision Models
1+
# Fine Tuning Torchvision Models
22

33
## Initialize the pretrained model
44

@@ -280,15 +280,22 @@ print('Predicted: ', ' '.join('%5s' % predicted[j]
280280

281281

282282
## Model Accuracy
283+
Please reference confusion_matrix.py and plot_confusion_matrix.py under Model_scripts, for more details on computing confusion matrix and model accuracy.
283284

284285
| Model | Epoch | Accuracy |
285286
|---------------|------ |------------ |
287+
| AlexNet | 30 | 72% |
286288
| ResNet 34 | 30 | 72% |
289+
| Mobilenet | 30 | 73% |
287290
| ResNet 50 | 30 | 75% |
288-
| AlexNet | 30 | 76% |
289-
| ResNet 101 | 30 | 77% |
290291
| densenet 161 | 30 | 77% |
292+
| ResNet 101 | 30 | 79% |
291293

294+
# Confusion Matrix for ResNet 101
295+
296+
The true positives rate and true negatives rate are both about 80%, which indicate a good precision and recall of our model.
297+
298+
![img](https://github.com/NaeRong/DS440_Capstone/blob/master/Images/cm.png)
292299

293300

294301

0 commit comments

Comments
 (0)