JLB-JLB's picture
Upload GAT/internal_only/training_config.json with huggingface_hub
58376e0 verified
{
"dataset_mode": "internal_only",
"test_ratio": 0.15,
"val_ratio": 0.15,
"random_seed": 42,
"model_type": "GAT",
"in_dim": 5,
"hidden_dim": 128,
"num_layers": 3,
"num_classes": 2,
"dropout": 0.5,
"batch_size": 4,
"learning_rate": 0.001,
"weight_decay": 0.0001,
"max_epochs": 200,
"es_patience": 20,
"es_min_epochs": 100,
"lr_patience": 10,
"lr_factor": 0.5,
"use_amp": true
}