| --- |
| language: |
| - en |
| metrics: |
| - accuracy |
| model-index: |
| - name: wild_cat_classification |
| results: |
| - task: |
| name: Image Classification |
| type: image-classification |
| dataset: |
| name: https://www.kaggle.com/datasets/gpiosenka/cats-in-the-wild-image-classification |
| type: imagefolder |
| config: default |
| split: train |
| args: default |
| metrics: |
| - name: Accuracy |
| type: accuracy |
| value: 0.9600 |
| --- |
| |
| # wild_cat_classification |
|
|
| This model is a fine-tuned version of vgg19 pre-trained model. |
| It achieves the following results on the evaluation set: |
| - Loss: 0.0095 |
| - Accuracy: 0.9600 |
|
|
| ### Training hyperparameters |
|
|
| The following hyperparameters were used during training: |
| - learning_rate: 1e-02 |
| - train_batch_size: 64 |
| - eval_batch_size: 64 |
| - optimizer: SGD |
| - num_epochs: 1 |
|
|
| ### Training results |
|
|
| | Training Loss | Epoch | Batch | |
| |:-------------:|:-----:|:----:| |
| | 0.0620 | 1.0 | 3/37 | |
| | 0.0425 | 1.0 | 6/37 | |
| | 0.0390 | 1.0 | 9/37 | |
| | 0.0258 | 1.0 | 12/37 | |
| | 0.0152 | 1.0 | 15/37 | |
| | 0.0180 | 1.0 | 18/37 | |
| | 0.0225 | 1.0 | 21/37 | |
| | 0.0129 | 1.0 | 24/37 | |
| | 0.0479 | 1.0 | 27/37 | |
| | 0.0231 | 1.0 | 30/37 | |
| | 0.0168 | 1.0 | 33/37 | |
| | 0.0095 | 1.0 | 36/37 | |
|
|
| ### Framework versions |
|
|
| - Pytorch 2.1.0+cu121 |
