Views
No views yet
| Parameter | Value |
|---|---|
| data_path | /home/felipe/hook_prompt_enhancer/training/data/safety_train.jsonl |
| data_splits | {'train': 6315, 'val': 701} |
| hyperparameters | {'learning_rate': 2e-05, 'batch_size': 16, 'epochs': 10, 'weight_decay': 0.01, 'warmup_ratio': 0.1, 'time_budget': 600} |
| model_name | microsoft/deberta-v3-base |
| seed | 42 |
| stage | safety |
| timestamp | 2026-03-29T11:59:45.399789+00:00 |
| val_macro_f1 | 0.9768 |
1import onnxruntime as ort
2from transformers import AutoTokenizer
3
4tokenizer = AutoTokenizer.from_pretrained("felipemunoz/prompt-enhancer-safety-deberta-v3")
5session = ort.InferenceSession("model.onnx")
6inputs = tokenizer("Is this prompt safe?", return_tensors="np")
7outputs = session.run(None, dict(inputs))