| { | |
| "model_type": "RetNet", | |
| "task": "text-classification", | |
| "architecture": "ProductionRetNet", | |
| "vocab_size": 50257, | |
| "model_dim": 512, | |
| "num_layers": 6, | |
| "num_heads": 8, | |
| "num_classes": 7, | |
| "max_length": 512, | |
| "labels": [ | |
| "EXPLICIT-DISCLAIMER", | |
| "EXPLICIT-OFFENSIVE", | |
| "EXPLICIT-SEXUAL", | |
| "EXPLICIT-VIOLENT", | |
| "NON-EXPLICIT", | |
| "SEXUAL-REFERENCE", | |
| "SUGGESTIVE" | |
| ], | |
| "label_to_id": { | |
| "EXPLICIT-DISCLAIMER": 0, | |
| "EXPLICIT-OFFENSIVE": 1, | |
| "EXPLICIT-SEXUAL": 2, | |
| "EXPLICIT-VIOLENT": 3, | |
| "NON-EXPLICIT": 4, | |
| "SEXUAL-REFERENCE": 5, | |
| "SUGGESTIVE": 6 | |
| }, | |
| "id_to_label": { | |
| "0": "EXPLICIT-DISCLAIMER", | |
| "1": "EXPLICIT-OFFENSIVE", | |
| "2": "EXPLICIT-SEXUAL", | |
| "3": "EXPLICIT-VIOLENT", | |
| "4": "NON-EXPLICIT", | |
| "5": "SEXUAL-REFERENCE", | |
| "6": "SUGGESTIVE" | |
| }, | |
| "tokenizer": "gpt2", | |
| "performance": { | |
| "holdout_accuracy": 0.7441, | |
| "holdout_macro_f1": 0.639, | |
| "inference_speed": "1574 paragraphs/sec", | |
| "parameters": 45029943 | |
| }, | |
| "training": { | |
| "dataset_size": 119023, | |
| "train_samples": 101771, | |
| "val_samples": 11304, | |
| "holdout_samples": 5948, | |
| "epochs": 5, | |
| "training_time_hours": 4.9, | |
| "focal_loss_gamma": 2.0 | |
| }, | |
| "model_file": "model.safetensors", | |
| "format": "safetensors" | |
| } |