linear classification

src1/linearclassification.py

@@ -35,8 +35,9 @@ class LinearClassification:
         lc = np.ones([train_features.shape[0], 1])
         X = np.concatenate((lc, train_features), axis=1)
         N = train_features.shape[1]
-        for k in range(1, len(set([label[0] for label in train_labels])) + 1):
-            train_label = linear_label([label[0] for label in train_labels], k)
+        tl = [label[0] for label in train_labels]
+        for k in range(1, len(set(tl)) + 1):
+            train_label = linear_label(tl, k)
             w = np.zeros([1, N + 1])
             for n in range(self.epochs):
                 y = np.dot(X, np.transpose(w))