YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
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
from model import StrongTextCNN
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐ Ask for provider support