GMARTINEZMILLA commited on
Commit
112b798
1 Parent(s): 6479a1d

Update utils.py

Browse files

imported CountVectorizer for it to work.

Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -2,7 +2,7 @@ import pandas as pd
2
  import numpy as np
3
  import warnings
4
  warnings.filterwarnings('ignore')
5
- from sklearn.feature_extraction.text import TfidfVectorizer
6
  from sklearn.metrics.pairwise import cosine_similarity
7
  from joblib import dump, load
8
  from sklearn.preprocessing import normalize
 
2
  import numpy as np
3
  import warnings
4
  warnings.filterwarnings('ignore')
5
+ from sklearn.feature_extraction.text import TfidfVectorizer, CountVectorizer
6
  from sklearn.metrics.pairwise import cosine_similarity
7
  from joblib import dump, load
8
  from sklearn.preprocessing import normalize