libcity 笔记：添加自定义dataset

• 假设我们把libcity/data/dataset/trajectory_dataset.py复制一份到libcity/data/dataset/dataset_subclass/GeolifeDM_dataset.py，里面内容不变，只是把class的名字换了

那其他需要修改哪些内容，使得这个dataset生效呢

•  libcity/data/dataset/dataset_subclass/__init__.py
  • 添加一行：from libcity.data.dataset.dataset_subclass.GeolifeDM_dataset import GeolifeDMDataset
  • 然后all列表添加GeolifeDMDataset

• /libcity/config/data/

  • 添加文件GeolifeDMDataset.json

  • 

• user_config/geolife_dm.json

  • 

执行即可

python run_model.py --task traj_loc_pred --model DeepMove --dataset gowalla  --config_file user_config/geolife_dm