RaghavIITM/enhanced-bilstm-nppe2

Enhanced BiLSTM Model for Named Entity Recognition

This model is a BiLSTM-based architecture for Named Entity Recognition tasks.

Model Details

• Architecture: Bidirectional LSTM
• Framework: PyTorch Lightning
• Checkpoint: epoch=24-step=2700.ckpt

Usage

import torch
from huggingface_hub import hf_hub_download

# Download checkpoint
checkpoint_path = hf_hub_download(
    repo_id="RaghavIITM/enhanced-bilstm-nppe2",
    filename="best_model.ckpt"
)

# Load model
checkpoint = torch.load(checkpoint_path, map_location='cpu')
print("Model loaded successfully!")

Training

• Trained on Kaggle
• Best checkpoint selected based on validation metrics

Links

• 🤗 Model on HuggingFace
• 📊 Kaggle Profile