From 57bb940cb82a54f5c8b2bbb3dc7820a025cb9eac Mon Sep 17 00:00:00 2001 From: Iwona Christop Date: Sat, 28 May 2022 17:57:38 +0200 Subject: [PATCH] Update rockyRoadtoDublin.py --- rockyRoadtoDublin.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/rockyRoadtoDublin.py b/rockyRoadtoDublin.py index 428e0a4..b252ac6 100644 --- a/rockyRoadtoDublin.py +++ b/rockyRoadtoDublin.py @@ -1,8 +1,12 @@ import inout as io def generateChooChoo(data, target, categories): - data = [x[2].replace('\n', '').replace(':', '') for x in io.read(data)] + data = io.read(data) + years = [x[0] for x in data] + text = [x[2].replace('\n', '').replace(':', '') for x in data] target = [x[0].replace('\n', '') for x in io.read(target)] + for i in range(len(text)): + data[i] = text[i] + ' year:' if categories == {}: i = 0 @@ -14,7 +18,11 @@ def generateChooChoo(data, target, categories): return {'data': data, 'target': target}, categories def predictFuture(test): - data = [x[2].replace('\n', '').replace(':', '') for x in io.read(test + '/in.tsv')] + data = io.read(test) + years = [x[0] for x in data] + text = [x[2].replace('\n', '').replace(':', '') for x in data] + for i in range(len(text)): + data[i] = text[i] + ' year:' with open('vw-' + test, 'w', encoding='utf-8') as f: for text in data: f.write('1 |text ' + text + '\n')