File size: 398 Bytes
0a3c8b0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
# StrongTextCNN
A CNN-based text classification model trained using PyTorch.
## Model Architecture
- Embedding Layer (300 dimensions)
- Conv1D kernels: 2, 3, 4, 5
- 128 filters each
- Batch Normalization
- Fully Connected Layers:
- 512 → 256
- 256 → 64
- 64 → 1
- Dropout = 0.4
## Files
- model.py
- best_model.pt
- tokenizer/
## Usage
```python
from model import StrongTextCNN
|