GraphRouter_eval/configs/config.yaml

# Config parameters for router data construction:
api_key: ""
llm_description_path: 'configs/new_LLM_Descriptions_with_think.json'
llm_embedding_path: 'configs/new_llm_description_embedding_with_think.pkl'
train_data_path: 'data/train.csv'
test_data_path: 'data/test.csv'
query_response_length: 512


# Config parameters for graph_router training:
seed: 2
wandb_key: ''
model_path: 'model_path/all_performance_best_model.pth'
train_epoch: 1000
scenario: "Performance First"  # "Performance First", "Balance", and "Cost First"
llm_num: 40
learning_rate: 0.0001
weight_decay: 0.0001
train_mask_rate: 0.5
batch_size: 64
num_task: 15
split_ratio: [0.9, 0.1]
embedding_dim: 8
edge_dim: 3