relbert/t_rex
Viewer • Updated • 1.59M • 719 • 3
How to use nikoslefkos/relex with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="nikoslefkos/relex") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("nikoslefkos/relex")
model = AutoModelForSequenceClassification.from_pretrained("nikoslefkos/relex")This model is a fine-tuned version of distilbert-base-cased on relbert/t_rex.Containing 291 labels for examples with more than 100 occurences. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
|---|---|---|---|---|
| 1.5343 | 0.6115 | 1.1212 | 0.6767 | 0 |
| 1.1175 | 0.6771 | 1.0503 | 0.6895 | 1 |
| 0.9745 | 0.7068 | 1.0405 | 0.6900 | 2 |
| 0.8598 | 0.7326 | 1.0456 | 0.6906 | 3 |
Base model
distilbert/distilbert-base-cased