Commit History
-
Add dataset JSON files
Aman | अमनa286609
-
Update README.md
Verified signaturebb87f91 -
Update README.md
Verified signaturec482d55 -
Create README.md
Verified signature# Task Classification Model (ICM) ## Model Description A BERT-based sequence classification model that routes computer vision questions to appropriate specialized modules. Classifies questions into 4 task categories: VQA, Captioning, Grounding, and Geometry. - **Repository:** beingamanforever/ICM - **Base Model:** bert-base-uncased - **Task:** 4-way Sequence Classification ## Labels | ID | Label | Description | |---|---|---| | 0 | vqa | Visual Question Answering ("What color is the car?") | | 1 | captioning | Image Description ("Describe the sunset.") | | 2 | grounding | Object Localization ("Find the person in the image.") | | 3 | geometry | Spatial/Metric Queries ("Calculate the area of the red box.") | ## Architecture BERT-Base encoder + 3-layer MLP classifier on [CLS] token: - Layer 1: Linear(768 → 256) + ReLU + Dropout(0.1) - Layer 2: Linear(256 → 128) + ReLU + Dropout(0.1) - Layer 3: Linear(128 → 4) ## Training | Hyperparameter | Value | |---|---| | Samples | 1,600 (4…2dba87c -
Upload folder using huggingface_hub
Verified signature392638a -
initial commit
Verified signaturede54b8d