You can deploy this model and start earning money today!

TrishanuDas/
CE-7_512_HuberLoss_train-31

Name: CE-7_512_HuberLoss_train-31
Availability: InStock

Views

No views yet

Usage

python

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

# Load tokenizer
tokenizer = AutoTokenizer.from_pretrained("TrishanuDas/CE-7_512_HuberLoss_train-31")

# Load model weights
model = AutoModelForSequenceClassification.from_pretrained("TrishanuDas/CE-7_512_HuberLoss_train-31")

# Prepare input
inputs = tokenizer("Query", "Document", return_tensors="pt", padding=True, truncation=True)

# Get prediction
with torch.no_grad():
    # Get logits
    outputs = model(**inputs)
    logits = outputs.logits
    
    # Apply sigmoid to get probabilities
    scores = torch.sigmoid(logits)

Important Note

When loading this model, you need to manually apply the sigmoid function to the logits as shown in the example above.