diff --git a/run.py b/run.py index 147748a..389b162 100644 --- a/run.py +++ b/run.py @@ -49,9 +49,7 @@ def evaluate_and_save(path, file, model, vectorizer): def main(): train_x = read_xz_file('train', 'in.tsv.xz', 500000) - # train_x = read_xz_file('train', 'in.tsv.xz') train_y = read_tsv('train', 'expected.tsv', 500000) - # train_y = read_tsv('train', 'expected.tsv') tfidf_vectorizer = TfidfVectorizer() train_x_vectorized = tfidf_vectorizer.fit_transform(train_x['col_name'].values)