Text Classification
Transformers
Safetensors
English
roberta
data-classification
sensitivity
data-governance
Synthetic
text-embeddings-inference
Instructions to use aitrainingdatabroker/document-sensitivity-tier-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use aitrainingdatabroker/document-sensitivity-tier-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="aitrainingdatabroker/document-sensitivity-tier-classifier")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("aitrainingdatabroker/document-sensitivity-tier-classifier") model = AutoModelForSequenceClassification.from_pretrained("aitrainingdatabroker/document-sensitivity-tier-classifier", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 504 Bytes
b19ec8c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | {
"metric": "accuracy and macro F1",
"accuracy": 0.831,
"macro_f1": 0.823,
"per_label": {
"confidential": {
"f1": 0.857,
"support": 19
},
"internal": {
"f1": 0.816,
"support": 20
},
"public": {
"f1": 0.974,
"support": 20
},
"restricted": {
"f1": 0.645,
"support": 18
}
},
"base_model": "distilroberta-base",
"train_rows": 305,
"test_rows": 77,
"epochs": 6,
"device": "cuda"
}
|