config.json

{
  "embed_dim": 64,
  "framework": "aprender",
  "hidden_dim": 128,
  "labels": [
    "safe",
    "needs-quoting",
    "non-deterministic",
    "non-idempotent",
    "unsafe"
  ],
  "max_seq_len": 64,
  "model_type": "shell-safety-classifier",
  "num_classes": 5,
  "training_data": "bashrs-corpus",
  "training_samples": 17942,
  "version": "2.0",
  "vocab_size": 251
}