dima806's picture
Update README.md
45b5b58 verified
metadata
license: apache-2.0
metrics:
  - accuracy
  - f1
base_model:
  - google/vit-base-patch16-224-in21k

Returns Tesla car model given an image with about 85% accuracy.

See https://www.kaggle.com/code/dima806/tesla-car-model-image-detection-vit for more details.

Classification report:

              precision    recall  f1-score   support

     Model_Y     0.8679    0.8364    0.8519        55
     Model_E     0.8462    0.8800    0.8627       100
     Model_S     0.8293    0.8095    0.8193        42
     Model_X     0.8519    0.8364    0.8440        55

    accuracy                         0.8492       252
   macro avg     0.8488    0.8406    0.8445       252
weighted avg     0.8493    0.8492    0.8490       252