Spaces:
Sleeping
Sleeping
Download src/nlp/embeddings.py from LeahRocks/SafeSpaceAI: direct link, hf CLI and curl.
- Browser
- Download file 241 Bytes
-
https://huggingface.co/spaces/LeahRocks/SafeSpaceAI/resolve/main/src/nlp/embeddings.py
- Command line
-
hf download hf://spaces/LeahRocks/SafeSpaceAI/src/nlp/embeddings.py
-
curl -L -o embeddings.py https://huggingface.co/spaces/LeahRocks/SafeSpaceAI/resolve/main/src/nlp/embeddings.py
241 Bytes
| from sklearn.feature_extraction.text import TfidfVectorizer | |
| vectorizer = TfidfVectorizer(max_features=5000) | |
| def fit_vectorizer(texts): | |
| return vectorizer.fit(texts) | |
| def transform(texts): | |
| return vectorizer.transform(texts).toarray() |