Update vit_model_test.py
Browse files- vit_model_test.py +1 -1
vit_model_test.py
CHANGED
|
@@ -11,7 +11,7 @@ from sklearn.metrics import accuracy_score, precision_score, confusion_matrix, f
|
|
| 11 |
import matplotlib.pyplot as plt
|
| 12 |
import seaborn as sns
|
| 13 |
from sklearn.metrics import recall_score
|
| 14 |
-
from
|
| 15 |
|
| 16 |
|
| 17 |
|
|
|
|
| 11 |
import matplotlib.pyplot as plt
|
| 12 |
import seaborn as sns
|
| 13 |
from sklearn.metrics import recall_score
|
| 14 |
+
from vit_model_training import labeling,CustomDataset
|
| 15 |
|
| 16 |
|
| 17 |
|