moreo
/
QuaPy
1
0
Fork 1
Code Issues Pull Requests Releases Wiki Activity

adding example with pre-trained classifier

This commit is contained in:
Alejandro Moreo Fernandez 2025-10-02 17:12:15 +02:00
parent b7931cf01a
commit 7a0cf2c5a2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/4.using_pretrained_classifier.py
View File

@ -14,7 +14,7 @@ import numpy as np
import quapy.functional as F import quapy.functional as F
# A scikit-learn's style wrapper for a huggingface-based pre-trained transformer # A scikit-learn's style wrapper for a huggingface-based pre-trained transformer for binary sentiment classification
class HFTextClassifier(BaseEstimator, ClassifierMixin): class HFTextClassifier(BaseEstimator, ClassifierMixin):
def __init__(self, model_name='distilbert-base-uncased-finetuned-sst-2-english'): def __init__(self, model_name='distilbert-base-uncased-finetuned-sst-2-english'):
self.pipe = pipeline("sentiment-analysis", model=model_name) self.pipe = pipeline("sentiment-analysis", model=model_name)