reshape data as suggested by sklearn

2017-08-24 12:01:09 -07:00
parent a2aa345712
commit 678517dd89
2 changed files with 2 additions and 2 deletions
--- a/talon/signature/extraction.py
+++ b/talon/signature/extraction.py
@@ -32,7 +32,7 @@ RE_REVERSE_SIGNATURE = re.compile(r'''

 def is_signature_line(line, sender, classifier):
    '''Checks if the line belongs to signature. Returns True or False.'''
-    data = numpy.array(build_pattern(line, features(sender)))
+    data = numpy.array(build_pattern(line, features(sender))).reshape(1, -1)
    return classifier.predict(data) > 0