python - 拟合函数返回 TypeError : float() argument must be string or a number in ScikitLearn

Question

我正在学习 scikit 学习执行某些分类。我正在按照我的数据集的教程进行操作。当我运行脚本时，出现类型错误

data = pd.DataFrame({'Description': pd.Categorical(["apple", "table", "red"]), 'Labels' : pd.Categorical(["Fruit","Furniture","Color"])})

counts = CountVectorizer().fit_transform(data['Description'].values)

tf_transformer = TfidfTransformer(use_idf=False).fit(counts)
train_tf = tf_transformer.transform(tf_transformer)

我得到的错误

Traceback (most recent call last):
  File "/anaconda/lib/python2.7/site-packages/IPython/core/interactiveshell.py", line 3035, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)
  File "<ipython-input-97-9a649172d3b7>", line 10, in <module>
    train_tf = tf_transformer.transform(tf_transformer)
  File "/anaconda/lib/python2.7/site-packages/sklearn/feature_extraction/text.py", line 1005, in transform
    X = sp.csr_matrix(X, dtype=np.float64, copy=copy)
  File "/anaconda/lib/python2.7/site-packages/scipy/sparse/compressed.py", line 69, in __init__
    self._set_self(self.__class__(coo_matrix(arg1, dtype=dtype)))
  File "/anaconda/lib/python2.7/site-packages/scipy/sparse/coo.py", line 204, in __init__
    self.data = self.data.astype(dtype)
TypeError: float() argument must be a string or a number

我一定是在做一些非常愚蠢的事情，因为我没有完全理解 API。有人可以告诉我如何解锁自己吗？

谢谢。

Answer 1

错误由此而来

tf_transformer.transform(tf_transformer)

我认为这是错误的语法 tf_transformer 是 TfidfTransformer 的对象。该函数需要稀疏矩阵。相反，您可以使用 fit_transform 函数

tf_transformer = TfidfTransformer(use_idf=False).fit_transform(counts)