在Python中将预测变量转换为分类

时间:2017-11-08 21:04:10

标签: python encoding keras categorical-data

Keras中是否有任何功能可以让我快速将数字/连续预测器转换为分类变量?我知道我们做了一个热门编码,但我在R中寻找类似的东西:

var = as.factor(var)

1 个答案:

答案 0 :(得分:0)

尝试:

from sklearn.preprocessing import LabelEncoder

encoder = LabelEncoder()
encoder.fit(var)
encoded_var = encoder.transform(var)

如果你想要一个热门编码:

dummy_var = np_utils.to_categorical(encoded_var)