metadata
license: mit
tags:
- image-classification
- document-classification
- pytorch
- computer-vision
datasets:
- tobacco3482
DocuSort AI
Document classification system trained on the Tobacco3482 dataset.
Models
- Custom CNN
- ResNet18 (Transfer Learning)
Dataset
Tobacco3482
Classes
- Advertisement
- Form
- Letter
- Memo
- News
- Note
- Report
- Resume
- Scientific
Performance
| Model | Test Accuracy | Macro-F1 |
|---|---|---|
| Custom CNN (from Scratch) | 67.5% | 0.684 |
| ResNet18 (Transfer Learning) | 70.6% | 0.706 |