From: rainfarmer <rainfarmer@16e7d947-dcc2-db11-b54a-0017319806e7>
Date: Wed, 17 Mar 2010 15:31:33 +0000 (+0000)
Subject: Probability output for L1R_LR
X-Git-Tag: v160~13
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=6ba93ef9e753f96eb6711676eec123dd1ce2c237;p=liblinear

Probability output for L1R_LR
---

diff --git a/linear.cpp b/linear.cpp
index 8509331..18aa226 100644
--- a/linear.cpp
+++ b/linear.cpp
@@ -1961,7 +1961,7 @@ int predict(const model *model_, const feature_node *x)
 
 int predict_probability(const struct model *model_, const struct feature_node *x, double* prob_estimates)
 {
-	if(model_->param.solver_type==L2R_LR)
+	if(model_->param.solver_type==L2R_LR || model_->param.solver_type==L1R_LR)
 	{
 		int i;
 		int nr_class=model_->nr_class;
diff --git a/predict.c b/predict.c
index 4a9d044..41c01e3 100644
--- a/predict.c
+++ b/predict.c
@@ -56,7 +56,7 @@ void do_predict(FILE *input, FILE *output, struct model* model_)
 	{
 		int *labels;
 
-		if(model_->param.solver_type!=L2R_LR)
+		if(model_->param.solver_type!=L2R_LR && model_->param.solver_type!=L1R_LR)
 		{
 			fprintf(stderr, "probability output is only supported for logistic regression\n");
 			exit(1);