Add decode.py

decode.py

@@ -0,0 +1,18 @@
+import inout as io
+
+
+if __name__ == '__main__':
+    target = [x[0].replace('\n', '') for x in io.read('train/expected.tsv.xz')]
+    categories = {}
+    i = 0
+    for x in target:
+        if x not in categories.values(): 
+            categories[i] = x
+            i += 1
+
+    files = ['dev-0', 'test-A', 'test-B']
+
+    for file in files:
+        predicted = io.read(file)
+        predicted = [categories[round(float(x))] for x in predicted]
+        io.write(predicted, file + '/out.tsv')

inout.py

@@ -22,4 +22,4 @@ def read(dir):
 def write(output, dir):
     with open(dir, 'w', newline='', encoding='utf-8') as f:
         writer = csv.writer(f)
-        writer.writerows(output)
+        for row in output: writer.writerow([row])