train_config.json

{
  "model_type": "bert",
  "pretrained": "bert-base-uncased",
  "epochs": 5,
  "batch_size": 16,
  "lr": 1e-05,
  "max_length": 512,
  "class_weights": [
    1.0,
    3.5319032669067383,
    10.121383666992188
  ],
  "num_labels": 3,
  "id2label": {
    "0": "SAME_PARAGRAPH",
    "1": "NEW_PARAGRAPH",
    "2": "NEWLINE"
  },
  "label2id": {
    "SAME_PARAGRAPH": 0,
    "NEW_PARAGRAPH": 1,
    "NEWLINE": 2
  }
}