Paths and setup
Train_h5
Val h_5
Hyperparameters
Num_classes=5
Batch_size=64
Epoch=50
LOAD FEATURE FUNCTION
Read Feature (X)
Read Labels(Y)
One hot Encoding Labels
Load Train/Val/Test Features and Build Classifier on top of ViT Features
Build and Compile Model
Setup Call backs
1.model checkpoints
2.Early stopping
3.Reduce Ron Plateau
Train Model with training data
Validate on X_Val every epoch and use callbacks
Evaluate model on test Data and analyse Accuracy
Test_h5
Val_h5
by nivi