Ошибка атрибута: нижний не найден

Я делаю классификацию документов и получил точность до 76%. И, предсказывая категорию документа, я сделал следующий

doc_clf.predict(tf_idf.transform((count_vect.transform([r'document']))))

и я получаю следующую ошибку:

File "/usr/local/lib/python3.5/dist- packages/sklearn/utils/metaestimators.py", line 115, in <lambda>
  out = lambda *args, **kwargs: self.fn(obj, *args, **kwargs)
File "/usr/local/lib/python3.5/dist-packages/sklearn/pipeline.py", line 306, in predict
  Xt = transform.transform(Xt)
File "/usr/local/lib/python3.5/dist-packages/sklearn/feature_extraction/text.py", line 923, in transform
  _, X = self._count_vocab(raw_documents, fixed_vocab=True)
File "/usr/local/lib/python3.5/dist-packages/sklearn/feature_extraction/text.py", line 792, in _count_vocab
  for feature in analyze(doc):
File "/usr/local/lib/python3.5/dist-packages/sklearn/feature_extraction/text.py", line 266, in <lambda>
  tokenize(preprocess(self.decode(doc))), stop_words)
File "/usr/local/lib/python3.5/dist-packages/sklearn/feature_extraction/text.py", line 232, in <lambda>
  return lambda x: strip_accents(x.lower())
File "/usr/local/lib/python3.5/dist-packages/scipy/sparse/base.py", line 647, in __getattr__
  raise AttributeError(attr + " not found")

Как я могу исправить эту ошибку? И есть ли еще способ повысить точность?

Я делюсь ссылкой, чтобы просмотреть полный код Полный код

1 ответ

Решение

В вашем коде doc_clf это трубопровод. Итак tf_idf.transform() а также count_vect.transform() будет обработан автоматически конвейером.

Вы должны только позвонить

category = doc_clf.predict([r'document'])

Когда этот документ проходит через конвейер, он будет автоматически преобразован CountVectorizer и TfidfTransformer.

Другие вопросы по тегам