From 93241f856790fd21dd93ae159644e4ee8dc55a26 Mon Sep 17 00:00:00 2001
From: Tomofumi NAKANO <Tomofumi_Nakano@voyagegroup.com>
Date: Mon, 11 Apr 2016 14:32:18 +0900
Subject: [PATCH] use softmax eq in predict_probability func

---
 linear.cpp | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/linear.cpp b/linear.cpp
index f9b9bdc..f557bd6 100644
--- a/linear.cpp
+++ b/linear.cpp
@@ -2685,10 +2685,13 @@ double predict_probability(const struct model *model_, const struct feature_node
 
 		double label=predict_values(model_, x, prob_estimates);
 		for(i=0;i<nr_w;i++)
-			prob_estimates[i]=1/(1+exp(-prob_estimates[i]));
+			prob_estimates[i]=exp(prob_estimates[i]);
 
 		if(nr_class==2) // for binary classification
+		{
+			prob_estimates[0]=1/(1+1/prob_estimates[0]);
 			prob_estimates[1]=1.-prob_estimates[0];
+		}
 		else
 		{
 			double sum=0;