|
|
--- |
|
|
library_name: onnx |
|
|
tags: |
|
|
- email-classification |
|
|
- job-search |
|
|
- onnx |
|
|
- transformers.js |
|
|
- browser-ml |
|
|
license: mit |
|
|
datasets: |
|
|
- custom |
|
|
language: |
|
|
- en |
|
|
pipeline_tag: text-classification |
|
|
--- |
|
|
|
|
|
# Email Classifier for Job Applications |
|
|
|
|
|
A fine-tuned BGE-small model that classifies emails into job application categories. Designed to run entirely in the browser using ONNX Runtime Web. |
|
|
|
|
|
## Model Description |
|
|
|
|
|
- **Base Model:** BAAI/bge-small-en-v1.5 |
|
|
- **Task:** 5-class email classification |
|
|
- **Format:** ONNX (optimized for browser inference) |
|
|
- **Size:** ~128MB |
|
|
|
|
|
## Labels |
|
|
|
|
|
| Label | Description | Application Status | |
|
|
|-------|-------------|-------------------| |
|
|
| `confirmation` | Application received/confirmed | Applied | |
|
|
| `rejection` | Application rejected | Rejected | |
|
|
| `interview` | Interview invitation | Interviewing | |
|
|
| `offer` | Job offer | Offer | |
|
|
| `not_job` | Not job-related | - | |
|
|
|
|
|
## Performance |
|
|
|
|
|
- **Validation Accuracy:** 99.65% |
|
|
- **Training Data:** 28,500 synthetic + curated emails |
|
|
|
|
|
## Usage with ONNX Runtime Web |
|
|
|
|
|
```javascript |
|
|
import * as ort from 'onnxruntime-web'; |
|
|
|
|
|
// Load model |
|
|
const session = await ort.InferenceSession.create( |
|
|
'https://huggingface.co/YOUR_USERNAME/email-classifier/resolve/main/model.onnx' |
|
|
); |
|
|
|
|
|
// Tokenize and run inference |
|
|
const results = await session.run({ |
|
|
input_ids: inputIdsTensor, |
|
|
attention_mask: attentionMaskTensor, |
|
|
}); |
|
|
``` |
|
|
|
|
|
## Files |
|
|
|
|
|
- `model.onnx` - The ONNX model file |
|
|
- `vocab.txt` - Vocabulary file for tokenization |
|
|
- `config.json` - Model configuration |
|
|
|
|
|
## Privacy |
|
|
|
|
|
This model runs 100% client-side in the browser. No email data is ever sent to a server. |
|
|
|
|
|
## License |
|
|
|
|
|
MIT |
|
|
|