Text Classifier

class TextClassifier

class factfinder.src.text_classifier.TextClassifier(repository_id='Sandrro/text_to_function_v2', number_of_categories=1, device_type=None)

Bases: object

This class is aimed to classify input texts into categories, or city functions. It uses a Huggingface transformer model trained on rubert-tiny

run(t)

This method takes a text as input and returns the predicted categories and probabilities. :param t: text to classify :return: list of predicted categories and probabilities