{
  "model_version": "V7",
  "model_name": "nlpaueb/legal-bert-base-uncased",
  "num_labels": 100,
  "label_names": [
    "Adjustments",
    "Agreements",
    "Amendments",
    "Anti-Corruption Laws",
    "Applicable Laws",
    "Approvals",
    "Arbitration",
    "Assignments",
    "Assigns",
    "Authority",
    "Authorizations",
    "Base Salary",
    "Benefits",
    "Binding Effects",
    "Books",
    "Brokers",
    "Capitalization",
    "Change In Control",
    "Closings",
    "Compliance With Laws",
    "Confidentiality",
    "Consent To Jurisdiction",
    "Consents",
    "Construction",
    "Cooperation",
    "Costs",
    "Counterparts",
    "Death",
    "Defined Terms",
    "Definitions",
    "Disability",
    "Disclosures",
    "Duties",
    "Effective Dates",
    "Effectiveness",
    "Employment",
    "Enforceability",
    "Enforcements",
    "Entire Agreements",
    "Erisa",
    "Existence",
    "Expenses",
    "Fees",
    "Financial Statements",
    "Forfeitures",
    "Further Assurances",
    "General",
    "Governing Laws",
    "Headings",
    "Indemnifications",
    "Indemnity",
    "Insurances",
    "Integration",
    "Intellectual Property",
    "Interests",
    "Interpretations",
    "Jurisdictions",
    "Liens",
    "Litigations",
    "Miscellaneous",
    "Modifications",
    "No Conflicts",
    "No Defaults",
    "No Waivers",
    "Non-Disparagement",
    "Notices",
    "Organizations",
    "Participations",
    "Payments",
    "Positions",
    "Powers",
    "Publicity",
    "Qualifications",
    "Records",
    "Releases",
    "Remedies",
    "Representations",
    "Sales",
    "Sanctions",
    "Severability",
    "Solvency",
    "Specific Performance",
    "Submission To Jurisdiction",
    "Subsidiaries",
    "Successors",
    "Survival",
    "Tax Withholdings",
    "Taxes",
    "Terminations",
    "Terms",
    "Titles",
    "Transactions With Affiliates",
    "Use Of Proceeds",
    "Vacations",
    "Venues",
    "Vesting",
    "Waiver Of Jury Trials",
    "Waivers",
    "Warranties",
    "Withholdings"
  ],
  "max_sequence_length": 256,
  "hidden_size": 768,
  "dropout_rate": 0.25,
  "training_gpus": 2,
  "test_metrics": {
    "classification": {
      "accuracy": 0.8793,
      "precision": 0.8757207303564448,
      "recall": 0.8793,
      "f1_score": 0.8749556050721212
    },
    "regression": {
      "r2_score": 0.8692134972569636,
      "mae": 3.5734300209999086,
      "rmse": 5.846707991168016
    },
    "risk_categories": {
      "accuracy": 0.9336
    }
  },
  "training_config": {
    "epochs": 15,
    "batch_size": 256,
    "batch_per_gpu": 128,
    "base_learning_rate": 4e-05,
    "warmup_ratio": 0.06,
    "loss_functions": {
      "classification": "CrossEntropyLoss (label_smoothing=0.1)",
      "regression": "Huber Loss (delta=0.1)"
    },
    "loss_weights": {
      "classification": 1.0,
      "regression": 1.0
    },
    "optimizer": "AdamW with LLRD (gamma=0.88)",
    "scheduler": "Cosine Annealing with Warmup",
    "risk_label_coverage": "100/100 classes"
  }
}