diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/gensvm_base.c | 1 | ||||
| -rw-r--r-- | src/gensvm_optimize.c | 3 |
2 files changed, 4 insertions, 0 deletions
diff --git a/src/gensvm_base.c b/src/gensvm_base.c index 6bb4367..10e4435 100644 --- a/src/gensvm_base.c +++ b/src/gensvm_base.c @@ -115,6 +115,7 @@ struct GenModel *gensvm_init_model() model->kerneltype = K_LINEAR; model->kernel_eigen_cutoff = 1e-8; model->max_iter = 1000000000; + model->elapsed_iter = -1; model->V = NULL; model->Vbar = NULL; diff --git a/src/gensvm_optimize.c b/src/gensvm_optimize.c index fff6768..b6bc164 100644 --- a/src/gensvm_optimize.c +++ b/src/gensvm_optimize.c @@ -123,6 +123,9 @@ void gensvm_optimize(struct GenModel *model, struct GenData *data) // store the training error in the model model->training_error = (Lbar - L)/L; + // store the iteration count in the model + model->elapsed_iter = it - 1; + // free the workspace gensvm_free_work(work); } |