R xgboost 将数字重新转换为字符

Question

我已经创建了一个 xgboost 模型并且对准确性感到满意。显然，我的 "Y" 现在是数字。有没有办法将我的预测值转换为原始字符？

Answer 1

是的，当我总结我的结果时，我通常会添加一个额外的原始标签列（即 "Placebo" 或 "Treatment" 而不是“0”或“1”）——你能调整这个吗满足您的需求？

Summary_of_results <- data.frame(Patient_ID = rownames(data),
                                 label = labels, 
                                 prediction = xgbpred)
Summary_of_results$original_label <- ifelse(
  Summary_of_results$label == 0,
  "Placebo",
  "Treatment")

Summary_of_results

R xgboost 将数字重新转换为字符

R xgboost re-convert numbers to character

r

xgboost