Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
psm1206 authored Jan 23, 2024
1 parent b9a451e commit 9dcddaa
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ conda env create -f mawu.yaml
conda activate mawu
```

### Run codes for DirectAU
### Run commands for DirectAU
```
python run_recbole.py --model=DirectAU --dataset=beauty --encoder=MF --weight_decay=1e-4 --gamma=0.4 &&
python run_recbole.py --model=DirectAU --dataset=beauty --encoder=LightGCN --weight_decay=1e-4 --gamma=0.4 &&
Expand All @@ -22,7 +22,7 @@ python run_recbole.py --model=DirectAU --dataset=yelp --encoder=MF --weight_deca
python run_recbole.py --model=DirectAU --dataset=yelp --encoder=LightGCN --weight_decay=1e-6 --gamma=2
```

### Run codes for MAWU
### Run commands for MAWU
```
python run_recbole.py --model=MAWU --dataset=beauty --encoder=MF --weight_decay=1e-4 --gamma1=1 --gamma2=0.1 &&
python run_recbole.py --model=MAWU --dataset=beauty --encoder=LightGCN --weight_decay=1e-4 --gamma1=0.9 --gamma2=0.2 &&
Expand Down

0 comments on commit 9dcddaa

Please sign in to comment.