Update README.md
Browse files
README.md
CHANGED
|
@@ -13,4 +13,10 @@ models:
|
|
| 13 |
- fsattar/brain-tumor-classifier
|
| 14 |
---
|
| 15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
| 13 |
- fsattar/brain-tumor-classifier
|
| 14 |
---
|
| 15 |
|
| 16 |
+
`brain_tumor_resnet34.ipynb` is the notebook I used to train the model
|
| 17 |
+
|
| 18 |
+
It's based on resnet34 (relatively lightweight model good at identifying a lot of things)
|
| 19 |
+
|
| 20 |
+
Using the base model, we fine tune with our dataset to narrow the model's answer set to just our classes
|
| 21 |
+
|
| 22 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|