diff options
Diffstat (limited to 'gensvm/cython_wrapper/wrapper.pyx')
| -rw-r--r-- | gensvm/cython_wrapper/wrapper.pyx | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/gensvm/cython_wrapper/wrapper.pyx b/gensvm/cython_wrapper/wrapper.pyx index e138620..a321102 100644 --- a/gensvm/cython_wrapper/wrapper.pyx +++ b/gensvm/cython_wrapper/wrapper.pyx @@ -179,6 +179,7 @@ def grid_wrap( int store_predictions, np.ndarray[np.int_t, ndim=1, mode='c'] cv_idx, int n_folds, + int verbosity, ): """ """ @@ -233,7 +234,8 @@ def grid_wrap( set_queue(queue, n_tasks, tasks) with nogil: - gensvm_train_q_helper(queue, cv_idx.data, store_predictions) + gensvm_train_q_helper(queue, cv_idx.data, store_predictions, + verbosity) cdef np.ndarray[np.int_t, ndim=1, mode='c'] pred |