diff --git a/run.py b/run.py index f832861..b9f2ac8 100644 --- a/run.py +++ b/run.py @@ -55,34 +55,4 @@ def main(): if __name__ == '__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) \ No newline at end of file + main() \ No newline at end of file