Modelpublic

marcev/financebert

sourceHugging Facegpl-3.0updated 2y agoView on Hugging Face

0likes176downloads

model_update.py9 linesDownload Raw Back to root

1from transformers import AutoModelForSequenceClassification2 3# Load your model4model = AutoModelForSequenceClassification.from_pretrained('path_to_your_local_model')5 6# Update label mapping7model.config.id2label = {0: 'Negative', 1: 'Neutral', 2: 'Positive'}8model.config.label2id = {'Negative': 0, 'Neutral': 1, 'Positive': 2}9