fix bug #91

Xtra-Computing · Jul 11, 2018 · bd0eb35 · bd0eb35
1 parent ee8f806
commit bd0eb35
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/thundersvmScikit.py b/python/thundersvmScikit.py
@@ -156,9 +156,9 @@ def _dense_fit(self, X, y, solver_type, kernel):
         else:
             weight_size = len(self.class_weight)
         weight_label = (c_int * weight_size)()
-        weight_label[:] = self.class_weight.keys()
+        weight_label[:] = list(self.class_weight.keys())
         weight = (c_float * weight_size)()
-        weight[:] = self.class_weight.values()
+        weight[:] = list(self.class_weight.values())
 
         n_features = (c_int * 1)()
         n_classes = (c_int * 1)()
@@ -198,9 +198,9 @@ def _sparse_fit(self, X, y, solver_type, kernel):
         else:
             weight_size = len(self.class_weight)
         weight_label = (c_int * weight_size)()
-        weight_label[:] = self.class_weight.keys()
+        weight_label[:] = list(self.class_weight.keys())
         weight = (c_float * weight_size)()
-        weight[:] = self.class_weight.values()
+        weight[:] = list(self.class_weight.values())
 
         n_features = (c_int * 1)()
         n_classes = (c_int * 1)()