s443930/ireland-news-headlines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update rockyRoadtoDublin.py

Browse Source
This commit is contained in:
Iwona Christop 2022-05-28 17:57:38 +02:00
parent 40653583a6
commit 57bb940cb8
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
rockyRoadtoDublin.py
View File

@ -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')