Skip to content

romaingrx/ViT-Tumor-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Tumor classification using Vision Transformers

After having trained the vision transformer for only 10 epochs, we can obtain an accuracy of 0.954 on the validation set. The evaluation can be seen on the image below.

evaluation.png