removed generate_embedding.sh . It was a test script and is not requi…

…red in the main repo
IITH-Compilers · Oct 30, 2023 · b1ad415 · b1ad415
1 parent 1647a56
commit b1ad415
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 28 deletions.
diff --git a/seed_embeddings/OpenKE/generate_embeddings.sh b/seed_embeddings/OpenKE/generate_embeddings.sh
diff --git a/seed_embeddings/OpenKE/preprocess.py b/seed_embeddings/OpenKE/preprocess.py
@@ -57,7 +57,7 @@ def getRelationDict(config):
     return relationDict
 
 
-def createTrain2ID(ed, rd, config):
+def createTrain2ID(entityDict, relationDict, config):
     ip = open(str(config.tripletFile), "r")
     content = ip.read()
     sentences = content.split("\n")
@@ -71,23 +71,42 @@ def createTrain2ID(ed, rd, config):
         l = len(s)
         if s[0] != "":
             if opc != "":
-                if s[0] not in ed:
+                if s[0] not in entityDict:
                     print(sentence)
                     print(s)
                     print(l)
                     print(str(sentences.index(sentence)))
                     print(s[0] + " not found in ed")
-                if "Next" not in rd:
-                    print("Next not found in rd")
-                toWrite += ed[opc] + "\t" + ed[s[0]] + "\t" + rd["Next"] + "\n"
+                if "Next" not in relationDict:
+                    print("Next not found in relationDict")
+                toWrite += (
+                    entityDict[opc]
+                    + "\t"
+                    + entityDict[s[0]]
+                    + "\t"
+                    + relationDict["Next"]
+                    + "\n"
+                )
                 nol += 1
             opc = s[0]
-            toWrite += ed[opc] + "\t" + ed[s[1]] + "\t" + rd["Type"] + "\n"
+            toWrite += (
+                entityDict[opc]
+                + "\t"
+                + entityDict[s[1]]
+                + "\t"
+                + relationDict["Type"]
+                + "\n"
+            )
             nol += 1
             i = 0
             for arg in range(2, l):
                 toWrite += (
-                    ed[opc] + "\t" + ed[s[arg]] + "\t" + rd["Arg" + str(i)] + "\n"
+                    entityDict[opc]
+                    + "\t"
+                    + entityDict[s[arg]]
+                    + "\t"
+                    + relationDict["Arg" + str(i)]
+                    + "\n"
                 )
                 nol += 1
                 i += 1