Slicky325's picture
Upload README.md with huggingface_hub
91d6bb6 verified
|
Raw
History Blame Contribute Delete
203 Bytes

Token Importance Classifier

Trained with F1: 0.9222, Accuracy: 0.8664

Usage

import torch
checkpoint = torch.load('selector.pt')
model.load_state_dict(checkpoint['model_state_dict'])