| --- |
| 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 |
| - Email |
| - 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 | |
