diff --git a/train.py b/train.py old mode 100755 new mode 100644 index 70cbda5..7788c7a --- a/train.py +++ b/train.py @@ -45,8 +45,8 @@ def predict_data(data): prepared = prepare_data(data) vectorized = vectorize.transform(data["text"]) predicted = bayes.predict_proba(vectorized)[:,1] - predicted[predicted < 0.05] = 0.05 - predicted[predicted > 0.95] = 0.95 + predicted[predicted < 0.05] = 0.00000005 + predicted[predicted > 0.95] = 0.99999995 return predicted dev_predicted = predict_data(dev_set)