aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGertjan van den Burg <gertjanvandenburg@gmail.com>2018-03-27 19:32:05 +0100
committerGertjan van den Burg <gertjanvandenburg@gmail.com>2018-03-27 19:32:05 +0100
commit25319a52f76a4b564efd01091818d7e464e72640 (patch)
tree23496d8ca115d4ac26acfb05dd653f880ff3a6c7
parentBugfix for reading kernel parameters from training files (diff)
downloadgensvm-25319a52f76a4b564efd01091818d7e464e72640.tar.gz
gensvm-25319a52f76a4b564efd01091818d7e464e72640.zip
Add command line flag to set the maximum number of iterations
-rw-r--r--src/GenSVMtraintest.c4
1 files changed, 4 insertions, 0 deletions
diff --git a/src/GenSVMtraintest.c b/src/GenSVMtraintest.c
index 63e6d58..284fe3d 100644
--- a/src/GenSVMtraintest.c
+++ b/src/GenSVMtraintest.c
@@ -76,6 +76,7 @@ void exit_with_help(char **argv)
printf("-g gamma : parameter for the rbf, polynomial or "
"sigmoid kernel\n");
printf("-h | -help : print this help.\n");
+ printf("-i max_iter : maximum number of iterations to do.\n");
printf("-k kappa : set the value of kappa used in the "
"Huber hinge (kappa > -1.0)\n");
printf("-l lambda : set the value of lambda "
@@ -302,6 +303,9 @@ void parse_command_line(int argc, char **argv, struct GenModel *model,
case 'g':
model->gamma = atof(argv[i]);
break;
+ case 'i':
+ model->max_iter = atoi(argv[i]);
+ break;
case 'k':
model->kappa = atof(argv[i]);
if (model->kappa <= -1.0)