Update tfidf.py
Browse files
tfidf.py
CHANGED
@@ -1,6 +1,5 @@
|
|
1 |
-
|
2 |
from sklearn.feature_extraction.text import TfidfVectorizer
|
3 |
tfidf = TfidfVectorizer(max_features=5000, ngram_range=(1, 2), stop_words='english')
|
4 |
X_train_tfidf = tfidf.fit_transform(X_train)
|
5 |
X_test_tfidf = tfidf.transform(X_test)
|
6 |
-
```
|
|
|
1 |
+
|
2 |
from sklearn.feature_extraction.text import TfidfVectorizer
|
3 |
tfidf = TfidfVectorizer(max_features=5000, ngram_range=(1, 2), stop_words='english')
|
4 |
X_train_tfidf = tfidf.fit_transform(X_train)
|
5 |
X_test_tfidf = tfidf.transform(X_test)
|
|