diff --git a/examples/4.using_pretrained_classifier.py b/examples/4.using_pretrained_classifier.py index 710048d..5b5ead5 100644 --- a/examples/4.using_pretrained_classifier.py +++ b/examples/4.using_pretrained_classifier.py @@ -14,7 +14,7 @@ import numpy as np 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): def __init__(self, model_name='distilbert-base-uncased-finetuned-sst-2-english'): self.pipe = pipeline("sentiment-analysis", model=model_name)