diff --git a/flair/embeddings/document.py b/flair/embeddings/document.py index 8f66a198ed..69fe9405b3 100644 --- a/flair/embeddings/document.py +++ b/flair/embeddings/document.py @@ -210,7 +210,7 @@ def embed(self, sentences: Union[list[Sentence], Sentence]): sentences = [sentences] raw_sentences = [s.to_original_text() for s in sentences] - tfidf_vectors = torch.from_numpy(self.vectorizer.transform(raw_sentences).A) + tfidf_vectors = torch.from_numpy(self.vectorizer.transform(raw_sentences).toarray()) for sentence_id, sentence in enumerate(sentences): sentence.set_embedding(self.name, tfidf_vectors[sentence_id])