BTX24
/

vit-base-patch16-224-in21k-finetuned-hongrui_mammogram_v_1

Image Classification

Generated from Trainer

Model card Files Files and versions Metrics Training metrics Community

BTX24 commited on Aug 31

Commit

3c56236

•

1 Parent(s): 0e42238

Update README.md

Files changed (1) hide show

README.md +5 -1

README.md CHANGED Viewed

@@ -11,6 +11,9 @@ metrics:
 model-index:
 - name: vit-base-patch16-224-in21k-finetuned-hongrui_mammogram_v_1
   results: []
 ---
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -70,10 +73,11 @@ The following hyperparameters were used during training:
 | 0.7087        | 9.0   | 1539 | 0.7445          | 0.6979   | 0.6697 | 0.6792    | 0.6979 |
 | 0.6977        | 10.0  | 1710 | 0.7419          | 0.6991   | 0.6767 | 0.6830    | 0.6991 |
 ### Framework versions
 - Transformers 4.42.4
 - Pytorch 2.4.0+cu121
 - Datasets 2.21.0
-- Tokenizers 0.19.1

 model-index:
 - name: vit-base-patch16-224-in21k-finetuned-hongrui_mammogram_v_1
   results: []
+datasets:
+- hongrui/mammogram_v_1
+pipeline_tag: image-classification
 ---
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 | 0.7087        | 9.0   | 1539 | 0.7445          | 0.6979   | 0.6697 | 0.6792    | 0.6979 |
 | 0.6977        | 10.0  | 1710 | 0.7419          | 0.6991   | 0.6767 | 0.6830    | 0.6991 |
+![Multi Class ROC Curve](https://cdn-uploads.huggingface.co/production/uploads/662eb39820de310d1558dd55/CToH2mNHE4zN6ihs3OqUz.png)
 ### Framework versions
 - Transformers 4.42.4
 - Pytorch 2.4.0+cu121
 - Datasets 2.21.0
+- Tokenizers 0.19.1