需求: 在做模型分类时,将label由字符串转换为数字
解决方案: label数据中共有20个label,提取这20个label,每一个换成指定数字
解决代码:
#--------Numberfy
str_to_num<-function(str_list){
num_list<-c(1:20)#可修改为指定数值
chr_list<-unique(Ytrain)#提取label
str_list$cuisine<-as.character(str_list$cuisine)#这里的cuisine为label
for(i in seq(1,length(chr_list))){
str_list$cuisine[test1$cuisine==chr_list[i]]<-num_list[i]
}
return(str_list)
}
由数字转换回字符串也是一个道理。但是用数字作为label所得到的预测结果会和设定的数值有一定偏差,需要根据情况进行二次处理。