Create tfidf.py
Browse files
tfidf.py
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
```python
|
| 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 |
+
```
|