diff --git a/articles/stackgbm.html b/articles/stackgbm.html index 713be0d..5db8ab1 100644 --- a/articles/stackgbm.html +++ b/articles/stackgbm.html @@ -289,18 +289,19 @@
Plot the ROC curves on the independent test set:
-pal <- c("#e69f00", "#56b4e9", "#009e73", "#f0e442")
+pal <- c("#e15759", "#f28e2c", "#59a14f", "#4e79a7", "#76b7b2")
-plot(smooth(roc_stack_te), col = pal[1])
-plot(smooth(roc_xgb_te), col = pal[2], add = TRUE)
-plot(smooth(roc_lgb_te), col = pal[3], add = TRUE)
-plot(smooth(roc_cat_te), col = pal[4], add = TRUE)
+plot(smooth(roc_stack_te), col = pal[1], lwd = 1)
+plot(smooth(roc_xgb_te), col = pal[2], lwd = 1, add = TRUE)
+plot(smooth(roc_lgb_te), col = pal[3], lwd = 1, add = TRUE)
+plot(smooth(roc_cat_te), col = pal[4], lwd = 1, add = TRUE)
legend(
"bottomright",
- col = pal, lwd = 2,
+ col = pal,
+ lwd = 2,
legend = c("stackgbm", "xgboost", "lightgbm", "catboost")
)