Spaces:
Sleeping
Sleeping
Update tasks/utils/predict.py
Browse files- tasks/utils/predict.py +2 -0
tasks/utils/predict.py
CHANGED
|
@@ -39,6 +39,8 @@ def predict(input_df: pd.DataFrame, tfidf_path:str , tfidf_voc_path:str, model_p
|
|
| 39 |
tfidf_vectorizer.set_params(preprocessor=process_text)
|
| 40 |
tfidf_vectorizer.set_params(vocabulary=vocab)
|
| 41 |
|
|
|
|
|
|
|
| 42 |
# Transform the input text using the TF-IDF vectorizer
|
| 43 |
text_data = input_df.to_pandas()["quote"]
|
| 44 |
text_features = tfidf_vectorizer.transform(text_data)
|
|
|
|
| 39 |
tfidf_vectorizer.set_params(preprocessor=process_text)
|
| 40 |
tfidf_vectorizer.set_params(vocabulary=vocab)
|
| 41 |
|
| 42 |
+
print(tfidf_vectorizer.vocabulary_)
|
| 43 |
+
|
| 44 |
# Transform the input text using the TF-IDF vectorizer
|
| 45 |
text_data = input_df.to_pandas()["quote"]
|
| 46 |
text_features = tfidf_vectorizer.transform(text_data)
|