metadata

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.96

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