forked from muhanzhang/IGMC
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathrun_transfer_exps.sh.orig
executable file
·36 lines (28 loc) · 1.59 KB
/
run_transfer_exps.sh.orig
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#! /bin/bash
# transfer models trained on ML-100K to the following datasets
data=${1}
num_relations=5
case ${data} in
yahoo_music)
multiply_by=20
;;
flixster)
multiply_by=1
;;
douban)
multiply_by=1
;;
*) echo 'Dataset does not exist.'
;;
esac
<<<<<<< HEAD
python Main.py --data-name ${data} --epochs 40 --testing --no-train --ensemble --transfer results/ml_100k_mnph200_testmode/ --dynamic-test --num-relations ${num_relations} --multiply-by ${multiply_by}
# for posterity, the original command is preserved below
# python Main.py --data-name ${data} --epochs 40 --testing --no-train --ensemble --transfer results/ml_100k_mnph200_testmode/ --dynamic-dataset --num-relations ${num_relations} --multiply-by ${multiply_by}
=======
# original line
# python Main.py --data-name ${data} --epochs 40 --testing --no-train --ensemble --transfer results/ml_100k_mnph200_testmode/ --dynamic-dataset --num-relations ${num_relations} --multiply-by ${multiply_by}
# new line
python Main.py --data-name ${data} --epochs 40 --testing --no-train --ensemble --transfer results/flixster_testmode/ --dynamic-test --num-relations ${num_relations} --multiply-by ${multiply_by}
>>>>>>> 0a6913f3b3ac89c572fafec9c4baebd5571f9fd2
# Note: The --transfer specifies where the ml_100k model to transfer is located. The --num-relations specifies how many edge types are there in the transferred model so that the target datasets can group their edge types to the same number. The --multiply-by specifies how many times to multiply the final prediction of the transferred model to accomodate different rating scales between source and target.