NaN check on input and making it so cache file is valid for smaller cache sizes.

1 files changed, 3 insertions, 1 deletions

diff --git a/main.cc b/main.cc
index e95bd411..1339d54d 100644
--- a/main.cc
+++ b/main.cc
@@ -23,7 +23,9 @@ int main(int argc, char *argv[]) {
 		cerr << endl << "weighted example sum = " << vars->weighted_examples;
 		cerr << endl << "weighted label sum = " << vars->weighted_labels;
 		cerr << endl << "average loss = " << vars->sum_loss / vars->weighted_examples;
-		cerr << endl << "best constant's loss = " << constant_loss;
+		cerr << endl << "best constant = " << best_constant;
+		if (vars->min_prediction == 0. && vars->max_prediction == 1. && best_constant < 1. && best_constant > 0.)
+		  cerr << endl << "best constant's loss = " << constant_loss;
 		cerr << endl << "total feature number = " << vars->total_features;
 		cerr << endl;
 	}