From 6ba93ef9e753f96eb6711676eec123dd1ce2c237 Mon Sep 17 00:00:00 2001 From: rainfarmer Date: Wed, 17 Mar 2010 15:31:33 +0000 Subject: [PATCH] Probability output for L1R_LR --- linear.cpp | 2 +- predict.c | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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); -- 2.50.1