richardschattner commited on
Commit
9c3b17b
·
verified ·
1 Parent(s): bc72147

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -8,7 +8,8 @@ metrics:
8
  This model is built on the efficientnet_b2 architecture.
9
  The model uses pretrained weights of the model found in the torchvision.models library.
10
  The classification head was changed, to be a dropout layer, followed by a linear layer with 6 target classes.
11
- Using transfer learning, the model was then trained on the [intel image dataset.](https://www.kaggle.com/datasets/puneet6060/intel-image-classification)
 
12
 
13
  #### Performance
14
  The model achieved a test accuracy of 89,67%.
 
8
  This model is built on the efficientnet_b2 architecture.
9
  The model uses pretrained weights of the model found in the torchvision.models library.
10
  The classification head was changed, to be a dropout layer, followed by a linear layer with 6 target classes.
11
+ Using transfer learning, the model was then trained on the [intel image dataset.](https://www.kaggle.com/datasets/puneet6060/intel-image-classification)
12
+ See the corresponding [hugging face space](https://huggingface.co/spaces/richardschattner/scenery_classification) for a live demo of the model.
13
 
14
  #### Performance
15
  The model achieved a test accuracy of 89,67%.