python - FeatureUnion Sklearn 管道中的错误

我有以下数据框:

ID Text
1  qwerty
2  asdfgh

我正在尝试为 文本 字段创建 md5 哈希并从上面的数据帧中删除 ID 字段。为了实现这一点，我创建了一个简单的 pipeline 和来自 sklearn 的自定义转换器。

这是我使用的代码:

class cust_txt_col(sklearn.base.BaseEstimator, sklearn.base.TransformerMixin):
    def __init__(self, key):
        self.key = key
    def fit(self, x, y=None):
        return self

    def hash_generate(self, txt):

        m = hashlib.md5()
        text = str(txt)
        long_text = ' '.join(text.split())
        m.update(long_text.encode('utf-8'))
        text_hash= m.hexdigest()
        return text_hash

    def transform(self, x):
        return x[self.key].apply(lambda  z: self.hash_generate(z)).values

class cust_regression_vals(sklearn.base.BaseEstimator, sklearn.base.TransformerMixin):
    def fit(self, x, y=None):
        return self
    def transform(self, x):
        x = x.drop(['Gene', 'Variation','ID','Text'], axis=1)
        return x.values

fp = pipeline.Pipeline([

 ('union', pipeline.FeatureUnion([
        ('hash', cust_txt_col('Text')), # can pass in either a pipeline
        ('normalized', cust_regression_vals()) # or a transformer
    ]))
])

当我运行它时，我收到以下错误:

ValueError: all the input arrays must have same number of dimensions

你能告诉我我的代码有什么问题吗？

如果我一一上课:

对于 cust_txt_col 我低于 o/p

['3e909f222a1e06098ec7ca1ea7e84540' '1691bdba3b75df145169e0501369fce3'
 '1691bdba3b75df145169e0501369fce3' ..., 'e11ec9863aaeb93f77a231319021e14d'
 '851c517b2af0a46cb9bc9373b748b6ff' '0ffe46fc75d21a5347b1f1a5a84526ad']

对于 cust_regression_vals 我低于 o/p

[[qwerty],
  [asdfgh]]

最佳答案

cust_txt_col 返回一个一维数组。 FeatureUnion 要求每个组成转换器返回一个二维数组。

关于python - FeatureUnion Sklearn 管道中的错误，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/45117902/