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