fixed batch training

YNedderhoff · Aug 9, 2015 · 63be3d9 · 63be3d9
1 parent 5140020
commit 63be3d9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tagger.py b/tagger.py
@@ -101,7 +101,7 @@ def train(self, file_in, file_out, max_iterations, top_x, decrease_alpha, shuffl
             # batch training:
             predictions = {}
             for tag in classifiers:
-                predictions[tag] = classifiers[tag].weight_vector
+                predictions[tag] = [x for x in classifiers[tag].weight_vector]
             print "\t\tEpoch " + str(i) + ", alpha = " + str(alpha)
             for ind, t in enumerate(tokens):
                 if ind % (len(tokens) / 10) == 0 and not ind == 0:
@@ -133,7 +133,7 @@ def train(self, file_in, file_out, max_iterations, top_x, decrease_alpha, shuffl
             # apply batch results to weight vectors:
             if batch_training:
                 for tag in classifiers:
-                    classifiers[tag].weight_vector = predictions[tag]
+                    classifiers[tag].weight_vector = [x for x in predictions[tag]]
 
             # decrease alpha
             if decrease_alpha: