Commit 78102c56 by Will Daly

Ngram changed back

parent 9803effd
......@@ -59,7 +59,7 @@ class ClassyAlgorithm(AIAlgorithm):
"""
pipeline = FeatureUnion([
('tfid', TfidfVectorizer(min_df=1, ngram_range=(1, 2), stop_words='english')),
('pos', CountVectorizer(tokenizer=tokenizer, ngram_range=(2, 3)))
('pos', CountVectorizer(tokenizer=tokenizer, ngram_range=(1, 2)))
])
transformed = pipeline.fit_transform([example.text for example in examples])
scores = [example.score for example in examples]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment