james-kramer commited on
Commit
573be17
·
verified ·
1 Parent(s): a778832

Update README.md

Browse files

Model Details
This model was trained on the aedupuga/cards-image-dataset for the task of image classification.

The model architecture used is timm_image with the resnet18 checkpoint, as determined by AutoGluon's AutoML process.

Training Details
The model was trained using AutoGluon's MultiModalPredictor with the following configuration:

Problem Type: Classification
Evaluation Metric: Accuracy
Presets: medium_quality
Hyperparameters:
model.names: ["timm_image"]
model.timm_image.checkpoint_name: "resnet18"
The training data used was the 'augmented' split of the dataset, with a 80/20 train/test split for tuning.

Evaluation Results
The model was evaluated on the original split of the dataset with an accuracy and weighted F1 of 1.00000.

Usage
You can load and use this model with AutoGluon.

Files changed (1) hide show
  1. README.md +7 -0
README.md CHANGED
@@ -1,3 +1,10 @@
 
 
 
 
 
 
 
1
  license: mit
2
  language:
3
  - en
 
1
+ ---
2
+ tags:
3
+ - autogluon
4
+ - multimodal
5
+ - image-classification
6
+ - resnet18
7
+ ---
8
  license: mit
9
  language:
10
  - en