From ad17f89c45c0eb4a5627a760a7fed914e693acb3 Mon Sep 17 00:00:00 2001 From: sapkotaruz11 Date: Mon, 27 Jan 2025 14:41:51 +0100 Subject: [PATCH] update literal training --- dicee/combined_literals_training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dicee/combined_literals_training.py b/dicee/combined_literals_training.py index fcf4d3a9..c6d15292 100644 --- a/dicee/combined_literals_training.py +++ b/dicee/combined_literals_training.py @@ -250,7 +250,7 @@ def forward(self, x, relation_idx, train_ent_embeddings=True): # Testing loop (denormalizing results and calculating metrics) test_df = pd.read_csv( - lit_dataset_dir, sep="\t", header=None, names=["head", "relation", "tail"] + lit_test_dir, sep="\t", header=None, names=["head", "relation", "tail"] ) test_df = test_df[test_df["relation"].isin(literal_dataset.train_rels)] test_df["head_idx"] = test_df["head"].map(literal_dataset.ent_idx)