-
Notifications
You must be signed in to change notification settings - Fork 1
/
run_all_models.sh
27 lines (23 loc) · 1.38 KB
/
run_all_models.sh
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
echo "Generating files for training."
python ../generate_dataset.py
echo "Got files for training."
ls ../pykg2vec/scene_data_1
ls ../pykg2vec/scene_data_2
ls ../pykg2vec/scene_data_3
echo "Starting TransE on dataset 1."
python ad_kge_train.py -mn TransE -ds ad_scene_relationships -dsp ~/Desktop/ad_kge/pykg2vec/pykg2vec/scene_data_1/
echo "Starting TransH on dataset 1."
python ad_kge_train.py -mn TransH -ds ad_scene_relationships -dsp ~/Desktop/ad_kge/pykg2vec/pykg2vec/scene_data_1/
echo "Starting Rescal on dataset 1."
python ad_kge_train.py -mn Rescal -ds ad_scene_relationships -dsp ~/Desktop/ad_kge/pykg2vec/pykg2vec/scene_data_1/
echo "Starting HoLE on dataset 1."
python ad_kge_train.py -mn HoLE -ds ad_scene_relationships -dsp ~/Desktop/ad_kge/pykg2vec/pykg2vec/scene_data_1/
echo "Starting TransE on dataset 2."
python ad_kge_train.py -mn TransE -ds ad_scene_relationships -dsp ~/Desktop/ad_kge/pykg2vec/pykg2vec/scene_data_2/
echo "Starting TransH on dataset 2."
python ad_kge_train.py -mn TransH -ds ad_scene_relationships -dsp ~/Desktop/ad_kge/pykg2vec/pykg2vec/scene_data_2/
echo "Starting Rescal on dataset 2."
python ad_kge_train.py -mn Rescal -ds ad_scene_relationships -dsp ~/Desktop/ad_kge/pykg2vec/pykg2vec/scene_data_2/
echo "Starting HoLE on dataset 2."
python ad_kge_train.py -mn HoLE -ds ad_scene_relationships -dsp ~/Desktop/ad_kge/pykg2vec/pykg2vec/scene_data_2/
echo "Done."