Skip to content

Commit

Permalink
update code and command
Browse files Browse the repository at this point in the history
  • Loading branch information
Shuyuan Xu committed Jun 22, 2023
1 parent ee9bade commit 2cee70f
Show file tree
Hide file tree
Showing 162 changed files with 4,353 additions and 330 deletions.
12 changes: 12 additions & 0 deletions .ipynb_checkpoints/environment-checkpoint.txt
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
python==3.9.7
transformers==4.26.0
torch==1.8.1+cu111
sklearn==1.1.2
torchvision==0.9.1+cu111
tqdm==4.64.1
time
collections
argparse
os
sys
numpy==1.23.1
22 changes: 22 additions & 0 deletions .ipynb_checkpoints/prompt-checkpoint.txt
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
sequential; seen; Considering {dataset} user_{user_id} has interacted with {dataset} items {history} . What is the next recommendation for the user ?; {dataset} {target}
sequential; seen; Here is the purchase history of {dataset} user_{user_id} : {dataset} item {history} . I wonder what is the next recommended item for the user .; {dataset} {target}
sequential; seen; {dataset} user_{user_id} has purchased {dataset} items {history}, predict next possible item to be bought by the user ?; {dataset} {target}
sequential; seen; I find the purchase list of {dataset} user_{user_id} : {dataset} items {history} , I wonder what other itmes does the user need . Can you help me decide ?; {dataset} {target}
sequential; seen; According to what items {dataset} user_{user_id} has purchased : {dataset} items {history} , Can you recommend another item to the user ?; {dataset} {target}
sequential; seen; What would {dataset} user_{user_id} be likely to purchase next after buying {dataset} items {history} ?; {dataset} {target}
sequential; seen; By analyzing the {dataset} user_{user_id} 's purchase of {dataset} items {history} , what is the next item expected to be bought ?; {dataset} {target}
sequential; seen; Can you recommend the next item for {dataset} user_{user_id} , given the user 's purchase of {dataset} items {history} ?; {dataset} {target}
sequential; seen; After buying {dataset} items {history} , what is the next item that could be recommended for {dataset} user_{user_id} ?; {dataset} {target}
sequential; seen; The {dataset} user_{user_id} has bought items : {dataset} items {history} , What else do you think is necessary for the user ?; {dataset} {target}
sequential; unseen; What is the top recommended item for {dataset} user_{user_id} who interacted with {dataset} item {history} ?; {dataset} {target}
straightforward; seen; What should we recommend for {dataset} user_{user_id} ?; {dataset} {target}
straightforward; seen; {dataset} user_{user_id} is looking for some items . Do you have any recommendations ?; {dataset} {target}
straightforward; seen; Do you have any suggested items for {dataset} user_{user_id} ?; {dataset} {target}
straightforward; seen; Which recommendation should we provide to {dataset} user_{user_id} ?; {dataset} {target}
straightforward; seen; How can we assist {dataset} user_{user_id} with a recommendation ?; {dataset} {target}
straightforward; seen; What would be a suitable recommendation for {dataset} user_{user_id} ?; {dataset} {target}
straightforward; seen; What would be a helpful recommendation for {dataset} user_{user_id} ?; {dataset} {target}
straightforward; seen; Can you recommend an item for {dataset} user_{user_id} ?; {dataset} {target}
straightforward; seen; Based on {dataset} user_{user_id} 's interests and requirements , what item would you suggest to try ?; {dataset} {target}
straightforward; seen; For {dataset} user_{user_id} , what item stands out as a top recommendation that they should consider ?; {dataset} {target}
straightforward; unseen; What is the top recommendation for {dataset} user_{user_id} ?; {dataset} {target}
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Beauty_collaborative-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Beauty --distributed 1 --gpu 0,1 --tasks sequential --item_indexing collaborative --epochs 20 --batch_size 64 --master_port 1112 --prompt_file ../prompt.txt --sample_prompt 0 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --collaborative_token_size 500 --collaborative_cluster 20 --test_filtered 0 --test_filtered_batch 0 --test_before_train 0 --test_epoch 2
2 changes: 2 additions & 0 deletions command/.ipynb_checkpoints/Beauty_random-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
python ../src/main.py --datasets Beauty --distributed 1 --gpu 0,1 --tasks sequential,straightforward --item_indexing random --epochs 20 --batch_size 64 --master_port 1995 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 1 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0

1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Beauty_sequential-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Beauty --distributed 1 --gpu 2,3 --tasks sequential,straightforward --item_indexing sequential --epochs 20 --batch_size 64 --master_port 2000 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 1 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/CDs_collaborative-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets CDs --distributed 1 --gpu 0,1 --tasks sequential,straightforward --item_indexing collaborative --epochs 10 --batch_size 128 --master_port 1994 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0 --collaborative_token_size 500 --collaborative_cluster 20
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/CDs_random-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets CDs --distributed 1 --gpu 4,5,6,7 --tasks sequential,straightforward --item_indexing random --epochs 10 --batch_size 128 --master_port 1994 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/CDs_sequential-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets CDs --distributed 1 --gpu 0,1,2,3 --tasks sequential,straightforward --item_indexing sequential --epochs 10 --batch_size 64 --master_port 1994 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Clothing_collaborative-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Clothing --distributed 1 --gpu 0,1 --tasks sequential,straightforward --item_indexing collaborative --epochs 10 --batch_size 128 --master_port 1024 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0 --collaborative_token_size 500 --collaborative_cluster 20
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Clothing_random-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Clothing --distributed 1 --gpu 0,1 --tasks sequential,straightforward --item_indexing random --epochs 10 --batch_size 128 --master_port 1024 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Clothing_sequential-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Clothing --distributed 1 --gpu 5,6 --tasks sequential,straightforward --item_indexing sequential --epochs 20 --batch_size 64 --master_port 1124 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Electronics_collaborative-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Electronics --distributed 1 --gpu 0,1 --tasks sequential,straightforward --item_indexing collaborative --epochs 10 --batch_size 128 --master_port 2008 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0 --collaborative_token_size 500 --collaborative_cluster 20
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Electronics_random-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Electronics --distributed 1 --gpu 0,1 --tasks sequential,straightforward --item_indexing random --epochs 10 --batch_size 256 --master_port 2008 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Electronics_sequential-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Electronics --distributed 1 --gpu 2,3 --tasks sequential,straightforward --item_indexing sequential --epochs 10 --batch_size 256 --master_port 2019 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/LastFM_collaborative-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets LastFM --distributed 1 --gpu 2,3 --tasks sequential,straightforward --item_indexing collaborative --epochs 20 --batch_size 64 --master_port 1995 --prompt_file ../prompt.txt --sample_prompt 0 --eval_batch_size 1 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 2 --collaborative_token_size 50 --collaborative_cluster 10
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/LastFM_random-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets LastFM --distributed 1 --gpu 4,5 --tasks sequential,straightforward --item_indexing random --epochs 20 --batch_size 64 --master_port 1995 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 1 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0 --collaborative_token_size 500 --collaborative_cluster 20 --collaborative_sparse 0 --collaborative_float32 0 --test_filtered 1 --test_filtered_batch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/LastFM_sequential-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets LastFM --distributed 1 --gpu 6,7 --tasks sequential,straightforward --item_indexing sequential --epochs 20 --batch_size 64 --master_port 1996 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 1 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0 --collaborative_token_size 500 --collaborative_cluster 20 --collaborative_sparse 0 --collaborative_float32 0 --test_filtered 1 --test_filtered_batch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/ML100K_collaborative-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets ML100K --distributed 1 --gpu 4,5 --tasks sequential,straightforward --item_indexing collaborative --epochs 10 --batch_size 128 --master_port 1994 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0 --collaborative_token_size 500 --collaborative_cluster 20
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/ML100K_random-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets ML100K --distributed 1 --gpu 6,7 --tasks sequential,straightforward --item_indexing random --epochs 20 --batch_size 64 --master_port 1994 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/ML100K_sequential-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets ML100K --distributed 1 --gpu 6,7 --tasks sequential,straightforward --item_indexing sequential --epochs 20 --batch_size 64 --master_port 1994 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/ML1M_collaborative-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets ML1M --distributed 1 --gpu 2,3 --tasks sequential,straightforward --item_indexing collaborative --epochs 10 --batch_size 128 --master_port 1991 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0 --collaborative_token_size 100 --collaborative_cluster 20
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/ML1M_random-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets ML1M --distributed 1 --gpu 0,1 --tasks sequential,straightforward --item_indexing random --epochs 10 --batch_size 128 --master_port 1991 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/ML1M_sequential-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets ML1M --distributed 1 --gpu 2,3 --tasks sequential,straightforward --item_indexing sequential --epochs 10 --batch_size 128 --master_port 1984 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Movies_collaborative-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Movies --distributed 1 --gpu 0,1,2,3 --tasks sequential,straightforward --item_indexing collaborative --epochs 10 --batch_size 128 --master_port 2008 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0 --collaborative_token_size 500 --collaborative_cluster 20
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Movies_random-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Movies --distributed 1 --gpu 0,1,2,3 --tasks sequential,straightforward --item_indexing random --epochs 10 --batch_size 128 --master_port 2008 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Movies_sequential-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Movies --distributed 1 --gpu 0,1,2,3 --tasks sequential,straightforward --item_indexing sequential --epochs 10 --batch_size 128 --master_port 2019 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/SP5_collaborative-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets ML1M,Yelp,LastFM,Beauty,ML100K,Clothing,CDs,Movies,Taobao,Electronics --distributed 1 --gpu 0,1,2,3 --tasks sequential,straightforward --item_indexing collaborative --epochs 10 --batch_size 128 --master_port 2017 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 10 --sample_num 1,1 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0 --collaborative_token_size 500 --collaborative_cluster 20
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/SP5_random-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets ML1M,Yelp,LastFM,Beauty,ML100K,Clothing,CDs,Movies,Taobao,Electronics --distributed 1 --gpu 0,1,2,3 --tasks sequential,straightforward --item_indexing random --epochs 10 --batch_size 256 --master_port 2017 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 10 --sample_num 1,1 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/SP5_sequential-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets ML1M,Yelp,LastFM,Beauty,ML100K,Clothing,CDs,Movies,Taobao,Electronics --distributed 1 --gpu 0,1,2,3 --tasks sequential,straightforward --item_indexing sequential --epochs 10 --batch_size 256 --master_port 2017 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 10 --sample_num 1,1 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Taobao_collaborative-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Taobao --distributed 1 --gpu 2,3 --tasks sequential,straightforward --item_indexing collaborative --epochs 10 --batch_size 128 --master_port 1994 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0 --collaborative_token_size 500 --collaborative_cluster 20
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Taobao_random-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Taobao --distributed 1 --gpu 0,1 --tasks sequential,straightforward --item_indexing random --epochs 10 --batch_size 128 --master_port 1994 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Taobao_sequential-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Taobao --distributed 1 --gpu 2,3 --tasks sequential,straightforward --item_indexing sequential --epochs 10 --batch_size 256 --master_port 1995 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Yelp_collaborative-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Yelp --distributed 1 --gpu 0,1,2,3 --tasks sequential,straightforward --item_indexing collaborative --epochs 10 --batch_size 128 --master_port 2019 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 2,2 --train 1 --test_prompt seen:0 --lr 1e-3 --collaborative_token_size 500 --test_before_train 0 --test_epoch 0 --collaborative_float32 1
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Yelp_random-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Yelp --distributed 1 --gpu 0,1,2,3 --tasks sequential,straightforward --item_indexing random --epochs 10 --batch_size 128 --master_port 2019 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 2,2 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
1 change: 1 addition & 0 deletions command/.ipynb_checkpoints/Yelp_sequential-checkpoint.sh
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
python ../src/main.py --datasets Yelp --distributed 1 --gpu 0,1,2,3 --tasks sequential,straightforward --item_indexing sequential --epochs 10 --batch_size 64 --master_port 2019 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 2,2 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
4 changes: 1 addition & 3 deletions command/Beauty_collaborative.sh
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1 @@
python ../src/main.py --datasets Beauty --distributed 1 --gpu 1,2 --tasks sequential,straightforward --item_indexing collaborative --epochs 20 --batch_size 64 --master_port 2023 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3

# wiser -r 9
python ../src/main.py --datasets Beauty --distributed 1 --gpu 0,1 --tasks sequential --item_indexing collaborative --epochs 20 --batch_size 64 --master_port 1112 --prompt_file ../prompt.txt --sample_prompt 0 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --collaborative_token_size 500 --collaborative_cluster 20 --test_filtered 0 --test_filtered_batch 0 --test_before_train 0 --test_epoch 2
3 changes: 1 addition & 2 deletions command/Beauty_random.sh
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,2 @@
python ../src/main.py --datasets Beauty --distributed 1 --gpu 3,4 --tasks sequential,straightforward --item_indexing random --epochs 20 --batch_size 64 --master_port 1994 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3
python ../src/main.py --datasets Beauty --distributed 1 --gpu 0,1 --tasks sequential,straightforward --item_indexing random --epochs 20 --batch_size 64 --master_port 1995 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 1 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0

# wiser -r 1646
4 changes: 1 addition & 3 deletions command/Beauty_sequential.sh
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1 @@
python ../src/main.py --datasets Beauty --distributed 1 --gpu 5,6 --tasks sequential,straightforward --item_indexing sequential --epochs 20 --batch_size 64 --master_port 2000 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 20 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3

# wiser -r 1647
python ../src/main.py --datasets Beauty --distributed 1 --gpu 2,3 --tasks sequential,straightforward --item_indexing sequential --epochs 20 --batch_size 64 --master_port 2000 --prompt_file ../prompt.txt --sample_prompt 1 --eval_batch_size 1 --dist_sampler 0 --max_his 20 --sample_num 3,3 --train 1 --test_prompt seen:0 --lr 1e-3 --test_before_train 0 --test_epoch 0
Loading

0 comments on commit 2cee70f

Please sign in to comment.