config.json

{
  "archi_style": "lo",
  "architectures": [
    "CodeBertE2EModel"
  ],
  "as_javadoc": true,
  "class_weights": [
    1.0,
    1.0
  ],
  "concat_test_first": false,
  "fnd_class_weights": [
    0.5013628189277285,
    183.94329896907217
  ],
  "fnd_clf_dropout": 0.1,
  "fnd_clf_hidden_size_1": 768,
  "fnd_clf_hidden_size_2": 512,
  "fnd_loss_fun": "bce",
  "fnd_one_line_text": false,
  "id2label": {
    "0": "NotRelated",
    "1": "Related"
  },
  "label2id": {
    "NotRelated": 0,
    "Related": 1
  },
  "lnk_class_weights": [
    0.5068125815335556,
    37.1968085106383
  ],
  "lnk_clf_dropout": 0.1,
  "lnk_clf_hidden_size_1": 512,
  "lnk_clf_hidden_size_2": 128,
  "lnk_loss_fun": "bce",
  "lnk_one_line_text": false,
  "loss_fun": "bce",
  "merge_embeddings": false,
  "model_type": "end2end",
  "suspect_threshold": 0.5,
  "torch_dtype": "float32",
  "transformers_version": "4.51.1",
  "use_cwe": false
}