Classification Models — README Template

Overview

This repo contains classification model implementations, training scripts, and example usage. All the models could be customized.

Repository layout

• src/
  • CNN.py # CNN (numbers of layers and hidden dimension)
  • MLP.py # MLP
  • VisualT.py # ViT (Rope, pre-trained weights)
  • ut.py # train function (minimal example)
  • EXP.ipynb # Example usage of: CNN, MLP, ViT, RF, XGBoost, Logit