在R中繪製模型的火車和評估分數

Question

如何在R中繪製列車和測試AUC打印在下方的分數（請參閱圖像）？

我使用下面

train.gdbt <- xgb.train(params=list(objective="binary:logistic", eval_metric="auc", eta=0.5, max_depth=4, subsample=1, colsample_bytree=0.5), data=dtrain, nrounds=500, watchlist=list(eval=dtest, train=dtrain),verbose = 1) 

Answer 1

xgb.train模型，您應該從xgb的evaluation_log，其修改爲長格式，然後繪製兩種顏色：

train.gdbt$evaluation_log %>% 
    gather(key=test_or_train, value=AUC, eval_auc, train_auc) %>% 
    ggplot(aes(x = iter, y = AUC, group = test_or_train, color = test_or_train)) + 
    geom_line() + 
    theme_bw()