diff --git a/vdiscover/Pipeline.py b/vdiscover/Pipeline.py index a428c9d..6a442c9 100644 --- a/vdiscover/Pipeline.py +++ b/vdiscover/Pipeline.py @@ -98,8 +98,8 @@ def make_train_pipeline(ftype): return Pipeline(steps=[ ('selector', ItemSelector(key='dynamic')), ('dvectorizer', CountVectorizer(tokenizer=dynamic_tokenizer, ngram_range=(1,3), lowercase=False, vocabulary=event_dict)), - #('todense', DenseTransformer()), - #('cutfoff', CutoffMax(16)), + ('todense', DenseTransformer()), + ('cutfoff', CutoffMax(16)), ('classifier', RandomForestClassifier(n_estimators=1000, max_features=None, max_depth=100)) #('classifier', GaussianNB())