additional datasets support

dynamic_model.py

@@ -25,17 +25,18 @@
 lastfm_simple = "lastfm_sim"
 lastfm3 = "lastfm3"
 instacart = "instacart"
+nowplaying = "nowplaying"
 
 #runtime settings
 flags = {}
 dataset = lastfm
 flags["context"] = True
 flags["temporal"] = True
-SEED = 1
-GPU = 1
+SEED = 2
+GPU = 0
 gap_strat = ""
 add = "_" if gap_strat != "" else ""
-directory = "/data/stud/bjorva/logs/sqrt/" + gap_strat + add
+directory = "/data/stud/bjorva/logs/dim/" + gap_strat + add
 debug = False
 
 torch.manual_seed(SEED)
@@ -62,7 +63,7 @@
 params["ALPHA"] = 0.45
 params["BETA"] = 0.45
 params["GAMMA"] = 0.1
-params["EPSILON"] = 0.5
+params["EPSILON"] = 1.0
 flags["use_day"] = True
 
 #data path and log/model-name
@@ -88,7 +89,7 @@
         MAX_EPOCHS -= 10
     min_time = 1.0
     flags["freeze"] = False
-elif dataset == lastfm or dataset == lastfm_simple or dataset == lastfm_time:
+elif dataset == lastfm or dataset == lastfm_simple or dataset == lastfm_time or dataset == nowplaying:
     dims["EMBEDDING_DIM"] = 100
     params["lr"] = 0.001
     params["dropout"] = 0.2

hawkes_baseline.py

@@ -12,7 +12,7 @@
 
 #global settings
 USE_DAY = True
-dataset = reddit_time
+dataset = lastfm
 n_decimals = 4
 
 #parameters
@@ -23,7 +23,7 @@
 
 #switchable
 full_hist = True
-gap_strat = ""
+gap_strat = "hawkes"
 
 add = "_" if gap_strat != "" else ""
 pickle_path = "hawkes_full_" + dataset + add + gap_strat + "4.pickle"

intra.py

@@ -13,11 +13,12 @@
 #datasets
 reddit = "subreddit"
 lastfm = "lastfm"
+nowplaying = "nowplaying"
 
 #set current dataset here
 dataset = lastfm
 dataset_path = "/data/stud/bjorva/datasets/" + dataset + "/4_train_test_split.pickle"
-pickle_path = "/data/stud/bjorva/logs/intra/"
+pickle_path = "/data/stud/bjorva/logs/nowplaying/"
 
 #universal settings
 BATCHSIZE = 100