File size: 351 Bytes
8e55ba7
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
{
  "train_data": "data/train/predicate_detection_data/llm_annotated/llm_annotated_benchmark_stratified_filtered.jsonl",
  "output_dir": "output/predicate_detector",
  "model_name": "bert-base-uncased",
  "val_ratio": 0.1,
  "num_epochs": 10,
  "batch_size": 16,
  "learning_rate": 5e-05,
  "seed": 42,
  "max_seq_length": 256,
  "debug_examples": 3
}