Hugginng face models about classy-classification HOT 7 CLOSED

Depends on the usecase and language.
However, you can also download my package fast-sentence-transformers and use it in combination with classy-classification. I will release this implementation later.

from classy_classification import classyClassifier

from fast_sentence_transformers import FastSentenceTransformer as SentenceTransformer

__all__ = ["SentenceTransformer"]


class FastClassyClassifier(classyClassifier):
    def __init__(self, embedding_model, *args, **kwargs):
        self.embedding_model = embedding_model
        super().__init__(*args, **kwargs)

    def set_embedding_model(self, model: str = None, device: str = "cpu"):
        self.model = model
        self.device = device
        pass

from classy-classification.

This should result in a x5 speedup on CPU.

from classy-classification.

@davidberenstein1957 appreciate your response.

from classy-classification.

@nsankar note that you should initialize the classifier with the fast_sentence_transformer embedding model.

from classy-classification.

Also, you can find a model overview here. https://www.sbert.net/docs/pretrained_models.html

from classy-classification.

Ok. Thanks

from classy-classification.

@nsankar I added the speed feature.

from classy-classification.