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