champion_challenger {DALEXtra} | R Documentation |
Compare machine learning models
Description
Determining if one model is better than the other one is a difficult task. Mostly because there is a lot of fields that have to be
covered to make such a judgement. Overall performance, performance on the crucial subset, distribution of residuals, those are only
few among many ideas related to that issue. Following function allow user to create a report based on various sections. Each says something different
about relation between champion and challengers. DALEXtra
package share 3 base sections which are funnel_measure
overall_comparison
and training_test_comparison
but any object that has generic plot
function can
be included at report.
Usage
champion_challenger(
sections,
dot_size = 4,
output_dir_path = getwd(),
output_name = "Report",
model_performance_table = FALSE,
title = "ChampionChallenger",
author = Sys.info()[["user"]],
...
)
Arguments
sections |
- list of sections to be attached to report. Could be sections available with DALEXtra which are |
dot_size |
- dot_size argument passed to |
output_dir_path |
- path to directory where Report should be created. By default it is current working directory. |
output_name |
- name of the Report. By default it is "Report" |
model_performance_table |
- If TRUE and |
title |
- Title for report, by default it is "ChampionChallenger". |
author |
- Author of , report. By default it is current user name. |
... |
- other parameters passed to rmarkdown::render. |
Value
rmarkdown report
Examples
library("mlr")
library("DALEXtra")
task <- mlr::makeRegrTask(
id = "R",
data = apartments,
target = "m2.price"
)
learner_lm <- mlr::makeLearner(
"regr.lm"
)
model_lm <- mlr::train(learner_lm, task)
explainer_lm <- explain_mlr(model_lm, apartmentsTest, apartmentsTest$m2.price, label = "LM")
learner_rf <- mlr::makeLearner(
"regr.ranger"
)
model_rf <- mlr::train(learner_rf, task)
explainer_rf <- explain_mlr(model_rf, apartmentsTest, apartmentsTest$m2.price, label = "RF")
learner_gbm <- mlr::makeLearner(
"regr.gbm"
)
model_gbm <- mlr::train(learner_gbm, task)
explainer_gbm <- explain_mlr(model_gbm, apartmentsTest, apartmentsTest$m2.price, label = "GBM")
plot_data <- funnel_measure(explainer_lm, list(explainer_rf, explainer_gbm),
nbins = 5, measure_function = DALEX::loss_root_mean_square)
champion_challenger(list(plot_data), dot_size = 3, output_dir_path = tempdir())