This commit is contained in:
Jakub Eichner 2022-06-07 17:10:37 +02:00
parent 2fc751b2f0
commit 624a9fbdb7

30
run.py
View File

@ -56,33 +56,3 @@ def main():
if __name__ == '__main__': if __name__ == '__main__':
main() main()
# testXPath = "./dev-0/in.tsv.xz"
# testYPath = "./dev-0/expected.tsv"
# testX = pd.read_csv(testXPath, sep='\t', nrows=nrows, header=None)
# testY = pd.read_csv(testYPath, sep='\t', nrows=nrows, header=None)
# testXtfidfVector = vectorizer.transform(testX[0])
# testXPath = "./test-A/in.tsv.xz"
# testYPath = "./test-A/expected.tsv"
# testX = pd.read_csv(testXPath, sep='\t', nrows=nrows, header=None)
# # testY = pd.read_csv(testYPath, sep='\t', nrows=nrows, header=None)
# testXtfidfVector = vectorizer.transform(testX[0])
# pred = gnb.predict(testXtfidfVector.todense())
# print(pred)
# with open(testYPath, 'w', newline='') as f_output:
# tsv_output = csv.writer(f_output, delimiter='\n')
# tsv_output.writerow(pred)