您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

在管道中创建虚拟模型,在训练和测试集中具有不同级别

在管道中创建虚拟模型,在训练和测试集中具有不同级别

您可以按照此答案中的说明使用类别:

categories = np.union1d(train, test)
train = train.astype('category', categories=categories)
test = test.astype('category', categories=categories)

pd.get_dummies(train)
Out: 
   a  b  c  d
0  1  0  0  0
1  0  1  0  0
2  0  1  0  0
3  1  0  0  0
4  1  0  0  0

pd.get_dummies(test)
Out: 
   a  b  c  d
0  1  0  0  0
1  0  1  0  0
2  0  0  1  0
3  0  0  0  1
其他 2022/1/1 18:52:12 有499人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶