1 files changed, 6 insertions, 7 deletions
diff --git a/run.py b/run.py
index 96fa10e..7b42ea8 100755
--- a/run.py
+++ b/run.py
@@ -5,7 +5,7 @@ import json
 import os
 from urllib.parse import quote_plus
 
-DATADIR = 'data'
+DATADIR = '/home/camil/temp/information-retrieval-data'
 ERRORFILE = 'errors'
 
 def get(dbpediaid):
@@ -21,7 +21,7 @@ def match(value, terms):
             return True
     return False
 
-def run(queries, line, outfile):
+def run(queries, line):
     query, _, dbpediaid, _, relevance, method = line.split('\t')
     terms = queries[query].split()
     try:
@@ -33,8 +33,8 @@ def run(queries, line, outfile):
             for value in values:
                 if match(value, terms):
                     matches += 1
-            outfile.write('{}\t{}\t{}\t{}\t{}\n'.format(
-                query, dbpediaid, field, len(values), matches))
+            print('{}\t{}\t{}\t{}\t{}\t{}\n'.format(
+                query, dbpediaid, relevance, field, len(values), matches))
     except Exception as e:
         print(dbpediaid)
         print(e)
@@ -45,6 +45,5 @@ if __name__ == '__main__':
     with open('queries_stopped.json') as f:
         queries = json.load(f)
 
-        with open('run.txt', 'w') as out:
-            for line in fileinput.input():
-                run(queries, line, out)
+        for line in fileinput.input():
+            run(queries, line)