def test_moco_hindex():
generate_emb(os.path.join(MOCO_PATH, "current.pth"), "h-index")
ret = run("h-index", "from_numpy", os.path.join(MOCO_PATH, "h-index.npy"))
assert ret["Micro-F1"] > 0.73, ret["Micro-F1"]
def test_moco_airport():
generate_emb(os.path.join(MOCO_PATH, "current.pth"), "usa_airport")
ret = run("usa_airport", "from_numpy",
os.path.join(MOCO_PATH, "usa_airport.npy"))
assert ret["Micro-F1"] > 0.63, ret["Micro-F1"]
def test_moco_rdt_5k():
NAME = "rdt-5k"
generate_emb(os.path.join(MOCO_PATH, "current.pth"), NAME)
ret = run(NAME, "from_numpy_graph", os.path.join(MOCO_PATH, f"{NAME}.npy"))
assert ret["Micro-F1"] > 0.52, ret
def test_moco_imdb_multi():
NAME = "imdb-multi"
generate_emb(os.path.join(MOCO_PATH, "current.pth"), NAME)
ret = run(NAME, "from_numpy_graph", os.path.join(MOCO_PATH, f"{NAME}.npy"))
assert ret["Micro-F1"] > 0.48, ret
def test_e2e_rdt_b():
NAME = "rdt-b"
generate_emb(os.path.join(E2E_PATH, "current.pth"), NAME)
ret = run(NAME, "from_numpy_graph", os.path.join(E2E_PATH, f"{NAME}.npy"))
assert ret["Micro-F1"] > 0.85, ret
def test_e2e_imdb_binary():
NAME = "imdb-binary"
generate_emb(os.path.join(E2E_PATH, "current.pth"), NAME)
ret = run(NAME, "from_numpy_graph", os.path.join(E2E_PATH, f"{NAME}.npy"))
assert ret["Micro-F1"] > 0.7, ret