Update svm_predict.py

svm_predict.py

@@ -18,8 +18,10 @@ def predict(sentence):
     for idx, word in enumerate(tokens):
         current_tag = sent_pos_tags[idx][1]
         current_idx = pos_tags.index(current_tag) if current_tag in pos_tags else -1
-        word_features = feature_vector(word, (1+idx)/l, current_idx)
+        prev_word = tokens[idx-1] if idx!=0 else ""
+        next_word = tokens[idx+1] if idx<l-1 else ""
+        word_features = feature_vector(word, (idx+1)/l, current_idx, prev_word, next_word)
         sent_features.append(word_features)
-        # scaled_features = scaler.transform(sent_features)
-    predictions = model.predict(sent_features)
+        scaled_features = scaler.transform(sent_features)
+    predictions = model.predict(scaled_features)
     return tokens, predictions