首页 > 解决方案 > 如何返回标签的原始值

问题描述

我正在使用排序过程,我必须将我的类转换为字符串以使用分解进行预测,如何将我的预测转换为我的文本

转换类参数 id

training['category_id'] = training['My Class'].factorize()[0]

运行预测后,如何在转换前记录源名称?

样本:

My Class
0 - Apple 
1 - Samsung
2 - HP 
3 - Motorola

在我的预测中返回一个数字,输入 3,我如何将我的代码转换为文本(摩托罗拉)?

标签: pythonpandasmachine-learningscikit-learn

解决方案


如果我理解正确,您希望在您的预测位置的数据框中获得“我的班级”标签。您可以使用iloc


推荐阅读