From 3ed51bb4ac5db6dfec92d10d6e302381c27849c8 Mon Sep 17 00:00:00 2001
From: Gertjan van den Burg <gertjanvandenburg@gmail.com>
Date: Wed, 6 Mar 2019 22:28:49 -0500
Subject: Add support for specifying sample weights (fixes #2)

---
 gensvm/cython_wrapper/wrapper.pyx | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'gensvm/cython_wrapper/wrapper.pyx')

diff --git a/gensvm/cython_wrapper/wrapper.pyx b/gensvm/cython_wrapper/wrapper.pyx
index 5c58fee..f98341b 100644
--- a/gensvm/cython_wrapper/wrapper.pyx
+++ b/gensvm/cython_wrapper/wrapper.pyx
@@ -33,6 +33,7 @@ def train_wrap(
         double kappa=0.0,
         double epsilon=1e-6,
         int weight_idx=1,
+        np.ndarray[np.float64_t, ndim=1, mode='c'] raw_weights=None,
         str kernel='linear',
         double gamma=1.0,
         double coef=0.0,
@@ -74,6 +75,9 @@ def train_wrap(
         gensvm_free_model(seed_model)
         seed_model = NULL
 
+    if not raw_weights is None:
+        set_raw_weights(model, raw_weights.data, n_obs)
+
     # Check the parameters
     error_msg = check_model(model)
     if error_msg:
-- 
cgit v1.2.3