diff options
Diffstat (limited to 'gensvm')
| -rw-r--r-- | gensvm/cython_wrapper/wrapper.pyx | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/gensvm/cython_wrapper/wrapper.pyx b/gensvm/cython_wrapper/wrapper.pyx index f98341b..0858103 100644 --- a/gensvm/cython_wrapper/wrapper.pyx +++ b/gensvm/cython_wrapper/wrapper.pyx @@ -82,6 +82,7 @@ def train_wrap( error_msg = check_model(model) if error_msg: gensvm_free_model(model) + gensvm_free_model(seed_model) free_data(data) error_repl = error_msg.decode('utf-8') raise ValueError(error_repl) @@ -106,6 +107,7 @@ def train_wrap( # free model and data gensvm_free_model(model); + gensvm_free_model(seed_model) free_data(data); return (V, n_SV, iter_count, training_error, fit_status) |