python ToG/main_wiki.py \
  --dataset gtsqa \
  --wikikg_dir ../data/ogbl-wikikg2 \
  --max_length 2048 \
  --temperature_exploration 0.4 \
  --temperature_reasoning 0 \
  --width 5 \
  --depth 4 \
  --remove_unnecessary_rel False \
  --LLM_type gpt-4o-mini \
  --num_retain_entity 5 \
  --prune_tools llm
