config.json

{
  "model_type": "radgraph_it",
  "architectures": [
    "RadgraphModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_radgraph.RadgraphConfig",
    "AutoModel": "modeling_radgraph.RadgraphModel"
  },
  "encoder_name": "IVN-RIN/medBIT-r3-plus",
  "max_length": 512,
  "max_span_width": 12,
  "feature_size": 20,
  "feedforward_params": {
    "hidden_dims": [
      150,
      150
    ],
    "dropout": 0.4
  },
  "loss_weights": {
    "ner": 0.2,
    "relation": 1.0
  },
  "relation_spans_per_word": 0.5,
  "train_encoder": true,
  "span_pooling": false,
  "transformer_params": null,
  "relation_context": false,
  "relation_feedforward_params": null,
  "dataset": "radgraph-it",
  "ner_labels": {
    "": 0,
    "Anatomy::definitely present": 1,
    "Observation::definitely present": 2,
    "Observation::definitely absent": 3,
    "Observation::uncertain": 4,
    "Anatomy::definitely absent": 5,
    "Anatomy::uncertain": 6
  },
  "relation_labels": {
    "": 0,
    "modify": 1,
    "located_at": 2,
    "suggestive_of": 3
  },
  "torch_dtype": "float32",
  "transformers_version": "5.14.1"
}