Librarian Bot: Add base_model information to model

#2
Files changed (1) hide show
  1. README.md +5 -4
README.md CHANGED
@@ -7,12 +7,14 @@ datasets:
7
  - keremberke/chest-xray-classification
8
  metrics:
9
  - accuracy
 
 
10
  model-index:
11
  - name: vit-xray-pneumonia-classification
12
  results:
13
  - task:
14
- name: Image Classification
15
  type: image-classification
 
16
  dataset:
17
  name: chest-xray-classification
18
  type: chest-xray-classification
@@ -20,10 +22,9 @@ model-index:
20
  split: validation
21
  args: full
22
  metrics:
23
- - name: Accuracy
24
- type: accuracy
25
  value: 0.9742489270386266
26
- pipeline_tag: image-classification
27
  ---
28
 
29
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
7
  - keremberke/chest-xray-classification
8
  metrics:
9
  - accuracy
10
+ pipeline_tag: image-classification
11
+ base_model: google/vit-base-patch16-224-in21k
12
  model-index:
13
  - name: vit-xray-pneumonia-classification
14
  results:
15
  - task:
 
16
  type: image-classification
17
+ name: Image Classification
18
  dataset:
19
  name: chest-xray-classification
20
  type: chest-xray-classification
 
22
  split: validation
23
  args: full
24
  metrics:
25
+ - type: accuracy
 
26
  value: 0.9742489270386266
27
+ name: Accuracy
28
  ---
29
 
30
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You