Integration of RE-GCN model

[JuliaGast]

Zixuan Li, Xiaolong Jin, Wei Li, Saiping Guan, Jiafeng Guo, Huawei Shen, Yuanzhuo Wang and Xueqi Cheng. Temporal Knowledge Graph Reasoning Based on Evolutional Representation Learning. SIGIR 2021.

RE-GCN

Tasks:

• [x] restructure code s.t. it fits tgb
• [x] implement data loading and make sure it is in correct format for CEN
• [x] integrate test()
• [x] store and load models at correct location
• [x] compute scores for negative samples only
• [x] compute and log mrr
• [x] compute and log runtime
• [ ] hyperparameter tuning
• [x] check if I can reproduce REGCN results from apples-oranges-paper

[JuliaGast]

YAGO:

    {
        "model": "REGCN",
        "data": "tkgl-yago",
        "run": 1,
        "seed": 1,
        "val mrr": 0.807044506072998,
        "test mrr": 0.818732738494873,
        "test_time": 167.84588862676173,
        "tot_train_val_time": 17007.569689787924
    },
    {
        "model": "REGCN",
        "data": "tkgl-yago",
        "run": 2,
        "seed": 2,
        "val mrr": 0.8092527389526367,
        "test mrr": 0.8198873400688171,
        "test_time": 167.5571665884927,
        "tot_train_val_time": 16749.494187387638
    },
    {
        "model": "REGCN",
        "data": "tkgl-yago",
        "run": 3,
        "seed": 3,
        "val mrr": 0.8115904927253723,
        "test mrr": 0.824786365032196,
        "test_time": 163.19445748906583,
        "tot_train_val_time": 16468.720900622196
    }

POLECAT-mini:

{
    "model": "REGCN",
    "data": "tkgl-polecat",
    "run": 1,
    "seed": 1,
    "val mrr": 0.16329748928546906,
    "test mrr": 0.1537366360425949,
    "test_time": 33.35347292479128,
    "tot_train_val_time": 2723.3249714439735
}

[JuliaGast]

done