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
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support