classification prediction plot colors inconsistent

[michaellevy]

They seem to switch if newdata is provided or not:

library(healthcareai)
m <- machine_learn(pima_diabetes, patient_id, outcome = diabetes, 
                   tune = FALSE, models = "xgb")
plot(predict(m))

plot(predict(m, pima_diabetes))

Created on 2018-08-27 by the reprex package (v0.2.0).