indra_bert_predicate_detector / training_config.json
thomaslim6793's picture
Update INDRA predicate detector
8e55ba7 verified
{
"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
}