Code style adjustments

This commit is contained in:
Jonas Winkler
2018-09-25 16:09:33 +02:00
parent 779ea6a015
commit 60618381f8
12 changed files with 94 additions and 81 deletions

2
src/documents/classifier.py Normal file → Executable file
View File

@@ -87,7 +87,7 @@ class DocumentClassifier(object):
# Step 2: vectorize data
logging.getLogger(__name__).info("Vectorizing data...")
self.data_vectorizer = CountVectorizer(analyzer='char', ngram_range=(3, 5), min_df=0.1)
self.data_vectorizer = CountVectorizer(analyzer="char", ngram_range=(3, 5), min_df=0.1)
data_vectorized = self.data_vectorizer.fit_transform(data)
self.tags_binarizer = MultiLabelBinarizer()