BehradG commited on
Commit
fa38903
1 Parent(s): 43e2665

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  library_name: transformers
3
- tags: []
4
  ---
5
 
6
  # Model Card for Model ID
@@ -77,13 +77,14 @@ Use the code below to get started with the model.
77
 
78
  ### Training Data
79
 
80
- <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
81
 
82
  [More Information Needed]
83
 
84
  ### Training Procedure
85
 
86
- <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
 
87
 
88
  #### Preprocessing [optional]
89
 
 
1
  ---
2
  library_name: transformers
3
+ tags: [Image Classification, MRI, Cancer]
4
  ---
5
 
6
  # Model Card for Model ID
 
77
 
78
  ### Training Data
79
 
80
+ https://huggingface.co/datasets/tanzuhuggingface/brainmri
81
 
82
  [More Information Needed]
83
 
84
  ### Training Procedure
85
 
86
+ The restnet18 model was fin-tuned with P100 GPU for 200 epochs. Both calibration and validation losses decined constantly during the fine-tuning showing no sign of overfitting.
87
+ The final accuracy was 97.9%.
88
 
89
  #### Preprocessing [optional]
90