update literal training

dice-group · Jan 27, 2025 · ad17f89 · ad17f89
1 parent 57fa021
commit ad17f89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)