我有一个带有 2 个分类变量、ID 变量和一个目标变量(用于分类)的 Pandas 数据框。我设法用OneHotEncoder
. 这导致稀疏矩阵。
ohe = OneHotEncoder()
# First I remapped the string values in the categorical variables to integers as OneHotEncoder needs integers as input
... remapping code ...
ohe.fit(df[['col_a', 'col_b']])
ohe.transform(df[['col_a', 'col_b']])
但我不知道如何在 DecisionTreeClassifier 中使用这个稀疏矩阵?特别是当我想稍后在我的数据框中添加一些其他非分类变量时。谢谢!
编辑 回复 miraculixx 的评论:我还尝试了 sklearn-pandas 中的 DataFrameMapper
mapper = DataFrameMapper([
('id_col', None),
('target_col', None),
(['col_a'], OneHotEncoder()),
(['col_b'], OneHotEncoder())
])
t = mapper.fit_transform(df)
但后来我得到这个错误:
TypeError:不支持类型转换:(dtype('O'),dtype('int64'),dtype('float64'),dtype('float64'))。