This project is organized in a HAFusion-style runtime layout while keeping the HRE model code unchanged. The New York experiment now uses HAFusion-style structural inputs: POI similarity, mobility adjacency, land-use similarity, and the original visual embedding.
data_NewYork/: New York dataset files used by the current projecttasks_NewYork/: downstream task scripts for the New York datasetproject_config.py: dataset and task registryparse_args.py: unified CLI parserHRE_Module_Train.py: training entry with dataset/task dispatchmain.py: simple launcher
Train for a single downstream task:
python main.py --city NewYork --task crime
python main.py --city NewYork --task checkin
python main.py --city NewYork --task servicecallRun all downstream tasks after training, while selecting the best checkpoint with a specific task:
python main.py --city NewYork --task all --selection_task crimeUseful runtime parameters:
python main.py --city NY --task crime --epochs 500 --learning_rate 0.0002 --dropout 0.2Outputs are saved under outputs/<city>/<task>/ by default.