Edit model card

Descriptive Sentences Classifier

Based on AlephBERT model.

Metrics

accuracy: 0.813953488372093
f1: 0.8181818181818182

How to Use the model:

from transformers import pipeline
classifier = pipeline("text-classification",model='orisuchy/Descriptive_Classifier', return_all_scores=True)
outputs = classifier("מסווג חתיך במיוחד")
print(outputs)

"""
Output:
[[
{'label': 'Descriptive', 'score': 0.999764621257782},
{'label': 'Not Descriptive', 'score': 0.00023541577684227377}]]
"""

Or, if you want only the final class:

from transformers import pipeline
classifier = pipeline("text-classification",model='orisuchy/Descriptive_Classifier')
output = classifier("הלכתי אליו הביתה וחיכיתי")
print(output)

"""
Output:
[{'label': 'Not Descriptive', 'score': 0.999901533126831}]
"""

Created by Daniel Smotritsky & Ori Suchy
GitHub

Downloads last month
90
Hosted inference API
Text Classification
Examples
Examples
This model can be loaded on the Inference API on-demand.

Dataset used to train orisuchy/Descriptive_Classifier