关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>
ChenGX
/
icdm_baseline_pyg
forked from GAMMALab/icdm_graph_competition
Not watched
1
0
Fork 0
Code
Releases 0 Wiki Activity
Issues 1 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
20 Commits
1 Branch
3.0 MiB
6 Download
 
 
Branch: master
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'master'
${ noResults }
icdm_baseline_pyg/pyg_example
ChenGX 2afa406f84 更新 'pyg_example/format_pyg.py' 1 year ago
..
README.md PyG update 1 year ago
format_pyg.py 更新 'pyg_example/format_pyg.py' 1 year ago
pyg_demo.sh PyG update 1 year ago
rgcn_sage_icdm.py PyG update 1 year ago

README.md

PyG example

Include

  • pyg_demo.sh: Shell Script (include converting raw data into PyG graph, training model and generating final result)
  • rgcn_mb_icdm.py: PyG rgcn model
  • format_pyg.py: PyG data generator
  • best_model/: Directory to store model trained

How to run

  • Run shell script
mkdir -p /dataset/pyg_data/
cd /code/icdm_graph_competition/pyg_example/
sh pyg_demo.sh

  • Output

    1. PyG Graph is stored in /dataset/pyg_data/

      • icdm2022_session1.pt PyG Heterograph of session1 data

      • icdm2022_session2.pt PyG Heterograph of session2 data

      • *.nodes.pyg Temporary cache file (Removable)

    2. Trained Model is saved in /code/icdm_graph_competition/best_model/

      • (Default) 1.pth Model file
      • (Optional) change model_id to customize name of model file

    3. Final result of Inference is generated in /code/icdm_graph_competition/pyg_example/

      • pyg_session1.json Result of session1
      • pyg_session2.json Result of session2
Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

No Description

Python Shell

Contributors (2)
All

Please read the following content carefully:

Dear OpenI User

Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.

For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》

Agree and continue
Disagree, exit