Skip to content

Latest commit

 

History

History
53 lines (40 loc) · 2.81 KB

README.md

File metadata and controls

53 lines (40 loc) · 2.81 KB

COMP4805 Project

This is a project using LightGCL(https://github.com/HKUDS/LightGCL) as the baseline model. We adopt denoising module to eliminate the noisy interactions in the original graph, as well as cross-layer contrastive loss and random noise to embedding.

Model Structure

denoiseGNN denoise_LightGCL_structure

Experiment results

Please refer to https://github.com/xcjames/comp4805/blob/main/log/summary.xlsx

Run the code and experiments

step1: clone the repository

git clone https://github.com/xcjames/comp4805.git

step2: extract the zip file tmall.zip in /data

step3: run the following python command

Experiments commands: Hyperparameter tuning:

python main.py --data yelp --add_noise_to_emb False --denoise False --cl_crossLayer_weight 0.05;
python main.py --data yelp --add_noise_to_emb False --denoise False --cl_crossLayer_weight 0.1;
python main.py --data yelp --add_noise_to_emb False --denoise False --cl_crossLayer_weight 0.2;
python main.py --data yelp --add_noise_to_emb False --denoise False --cl_crossLayer_weight 0.3;
python main.py --data yelp --add_noise_to_emb False --denoise False --cl_crossLayer_weight 0.5;

python main.py --data yelp --add_noise_to_emb False --cl_crossLayer False --beta 0.02;
python main.py --data yelp --add_noise_to_emb False --cl_crossLayer False --beta 0.05;
python main.py --data yelp --add_noise_to_emb False --cl_crossLayer False --beta 0.1;
python main.py --data yelp --add_noise_to_emb False --cl_crossLayer False --beta 0.2;

python main.py --data yelp --denoise False --cl_crossLayer False --eps 0.01;
python main.py --data yelp --denoise False --cl_crossLayer False --eps 0.025;
python main.py --data yelp --denoise False --cl_crossLayer False --eps 0.05;
python main.py --data yelp --denoise False --cl_crossLayer False --eps 0.1;

On different datasets:

python main.py --data yelp --add_noise_to_emb False --denoise False --cl_crossLayer False;
python main.py --data yelp --eps 0.025 --beta 0.05 --cl_crossLayer_weight 0.1;

python main.py --data gowalla --add_noise_to_emb False --denoise False --cl_crossLayer False;
python main.py --data gowalla --eps 0.025 --beta 0.05 --cl_crossLayer_weight 0.1;
python main.py --data gowalla --eps 0.025 --beta 0.1 --cl_crossLayer_weight 0.2;

python main.py --data tmall --add_noise_to_emb False --denoise False --cl_crossLayer False --epoch 30;
python main.py --data tmall --eps 0.025 --beta 0.05 --cl_crossLayer_weight 0.1 --epoch 30;
python main.py --data tmall --eps 0.025 --beta 0.1 --cl_crossLayer_weight 0.2 --epoch 30;