Added breast_cancer dataset

makgyver · Mar 7, 2019 · 9e69741 · 9e69741
1 parent a47703d
commit 9e69741
Show file tree

Hide file tree

Showing 2 changed files with 684 additions and 1 deletion.
diff --git a/PRL/prl.py b/PRL/prl.py
@@ -1,6 +1,7 @@
 import math
 import numpy as np
 
+from .genK import *
 from .genF import *
 from .genP import *
 
@@ -205,7 +206,7 @@ def predict(self, gen_pref_test):
         for i in range(gen_pref_test.n):
             x = X[i,:]
             sco = [0.0 for c in range(self.dim)]
-            for j, (p, k) in enumerate(self.col_list):
+            for j, (p, f) in enumerate(self.col_list):
                 if self.Q[j] > 0.0:
                     for c in range(self.dim):
                         if p[0][1] == c: