Probability output for L1R_LR

diff --git a/linear.cpp b/linear.cpp
index 8509331939e7415fcb8f9500224bfe23416d866d..18aa2268d43dc8288aa331bc8eba1e0db6c45edb 100644 (file)
--- 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 4a9d044761e55a9a76e95e6613bc530e62888502..41c01e371db6ba358c538de1799dd7f70897ffc8 100644 (file)
--- 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);