From 6666f0baf8d02c331993dd67317829169f4ab8a1 Mon Sep 17 00:00:00 2001
From: Al <albarrentine@gmail.com>
Date: Tue, 29 Sep 2015 13:06:34 -0400
Subject: [PATCH] [fix] Labels in averaged perceptron tagger

---
 src/averaged_perceptron_tagger.c | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/averaged_perceptron_tagger.c b/src/averaged_perceptron_tagger.c
index 2807dde9..83c7324a 100644
--- a/src/averaged_perceptron_tagger.c
+++ b/src/averaged_perceptron_tagger.c
@@ -17,13 +17,17 @@ bool averaged_perceptron_tagger_predict(averaged_perceptron_t *model, void *tagg
         cstring_array_clear(features);
 
         if (i > 0) {
-            prev = cstring_array_get_string(labels, prev_id);
+            prev = cstring_array_get_string(model->classes, prev_id);
         }
 
         if (i > 1) {
-            prev2 = cstring_array_get_string(labels, prev2_id);            
+            prev2 = cstring_array_get_string(model->classes, prev2_id);            
+        } else if (i == 1) {
+            prev2 = START;
         }
 
+        log_debug("prev=%s, prev2=%s\n", prev, prev2);
+
         if (!feature_function(tagger, context, tokenized, i, prev, prev2)) {
             log_error("Could not add address parser features\n");
             return false;