fixed batch training

YNedderhoff · Aug 9, 2015 · 13d2821 · 13d2821
1 parent 5f4709f
commit 13d2821
Show file tree

Hide file tree

Showing 6 changed files with 39,990 additions and 2 deletions.
diff --git a/named-entity-recognizer.py b/named-entity-recognizer.py
@@ -133,7 +133,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)
             path = []
             normalization_constant = 10.0 ** 5.0
@@ -253,7 +253,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: