From 75182ca28c2b481420fe42e6fc469075ec6b4cdf Mon Sep 17 00:00:00 2001
From: Gertjan van den Burg <gertjanvandenburg@gmail.com>
Date: Tue, 21 Feb 2017 18:32:35 -0500
Subject: Keep a status of training

---
 src/gensvm_optimize.c | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'src/gensvm_optimize.c')

diff --git a/src/gensvm_optimize.c b/src/gensvm_optimize.c
index b6bc164..df46ec4 100644
--- a/src/gensvm_optimize.c
+++ b/src/gensvm_optimize.c
@@ -104,13 +104,21 @@ void gensvm_optimize(struct GenModel *model, struct GenData *data)
 		it++;
 	}
 
+	// status == 0 means training was successful
+	model->status = 0;
+
 	// print warnings if necessary
-	if (L > Lbar)
+	if (L > Lbar) {
 		err("[GenSVM Warning]: Negative step occurred in "
 				"majorization.\n");
-	if (it >= model->max_iter)
+		model->status = 1;
+	}
+
+	if (it >= model->max_iter) {
 		err("[GenSVM Warning]: maximum number of iterations "
 				"reached.\n");
+		model->status = 2;
+	}
 
 	// print final iteration count and loss
 	note("Optimization finished, iter = %li, loss = %15.16f, "
-- 
cgit v1.2.3