diff --git a/predict_rm.py b/predict_rm.py index 95762ba..6bface1 100644 --- a/predict_rm.py +++ b/predict_rm.py @@ -18,7 +18,7 @@ def predict(): y_dev = clf.predict_proba(dev0_vectorizer)[:, 1] y_test = clf.predict_proba(testA_vectorizer)[:, 1] - np.savetxt('test-A/out.tsv', y_dev, '%f') - np.savetxt('dev-0/out.tsv', y_test, '%f') + np.savetxt('test-A/out.tsv', y_test, '%f') + np.savetxt('dev-0/out.tsv', y_dev, '%f') predict()