allow max_iter to be set in the model

2 files changed, 4 insertions, 0 deletions

diff --git a/include/gensvm_base.h b/include/gensvm_base.h
index dee1d80..adef2a9 100644
--- a/include/gensvm_base.h
+++ b/include/gensvm_base.h
@@ -136,6 +136,8 @@ struct GenModel {
 	double kernel_eigen_cutoff;
 	///< cutoff value for the ratio of eigenvalues in the reduced 
 	//eigendecomposition.
+	long max_iter;
+	///< maximum number of iterations of the algorithm
 };
 
 /**
diff --git a/include/gensvm_task.h b/include/gensvm_task.h
index a791262..77db463 100644
--- a/include/gensvm_task.h
+++ b/include/gensvm_task.h
@@ -75,6 +75,8 @@ struct GenTask {
 	///< coef parameter for the GenModel
 	double degree;
 	///< degree parameter for the GenModel
+	long max_iter;
+	///< maximum number of iterations of the algorithm
 	struct GenData *train_data;
 	///< pointer to the training data
 	struct GenData *test_data;