Commit 78102c56 by Will Daly

Ngram changed back

parent 9803effd
...@@ -59,7 +59,7 @@ class ClassyAlgorithm(AIAlgorithm): ...@@ -59,7 +59,7 @@ class ClassyAlgorithm(AIAlgorithm):
""" """
pipeline = FeatureUnion([ pipeline = FeatureUnion([
('tfid', TfidfVectorizer(min_df=1, ngram_range=(1, 2), stop_words='english')), ('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]) transformed = pipeline.fit_transform([example.text for example in examples])
scores = [example.score 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