config.json

{
  "model_type": "wrag2",
  "architecture": "WRAG2TextModel",
  "base_model": "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
  "num_shards": 10,
  "k": 3,
  "num_wr_layers": 3,
  "hidden_size": 2048,
  "num_classes": 2,
  "task": "text-classification",
  "domains": [
    "medical",
    "legal",
    "code"
  ],
  "version": "1.0.0",
  "results": {
    "medical_accuracy": 0.7,
    "legal_accuracy": 0.84,
    "code_accuracy": 0.95,
    "average_accuracy": 0.83
  }
}