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

Commit f4fcb89

Browse files
authored
Update Fine Tuning Torchvision Models.md
Add links to models
1 parent d12bf1a commit f4fcb89

1 file changed

Lines changed: 13 additions & 1 deletion

File tree

Tutorials/Fine Tuning Torchvision Models.md

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -291,13 +291,25 @@ Please reference confusion_matrix.py and plot_confusion_matrix.py under Model_sc
291291
| densenet 161 | 30 | 77% |
292292
| ResNet 101 | 30 | 79% |
293293

294+
### AlexNet model
295+
[This link](https://github.com/NaeRong/DS440_Capstone/blob/master/master/Tutorials/Model%20Script/alexnet.py)
296+
297+
### ResNet 34 model
298+
[This link](https://github.com/NaeRong/DS440_Capstone/blob/master/master/Tutorials/Model%20Script/resnset34.py)
299+
300+
### ResNet 101 model
301+
[This link](https://github.com/NaeRong/DS440_Capstone/blob/master/master/Tutorials/Model%20Script/resnset101.py)
302+
294303
## Confusion Matrix for ResNet 101
295304

296305
The true positives rate and true negatives rate are both about 80%, which indicate a good precision and recall of our model.
297306

298307
![img](https://github.com/NaeRong/DS440_Capstone/blob/master/Images/cm.png)
299308

300-
309+
### Script to compute Confusion Matrix
310+
[This link](https://github.com/NaeRong/DS440_Capstone/blob/master/master/Tutorials/Model%20Script/Confusion_Matrix.py)
311+
### Script to compute Confusion Matrix Plot
312+
[This link](https://github.com/NaeRong/DS440_Capstone/blob/master/master/Tutorials/Model%20Script/Plot_Confusion_Matrix.py)
301313

302314
## License
303315

0 commit comments

Comments
 (0)