fstermann / mlr-mini

MIT License
0 stars 2 forks source link

Performance Evaluators #8

Open m-muecke opened 1 year ago

m-muecke commented 1 year ago

Tasks

Description

mlr.mini should provide a few functions that evaluate the performance of a prediction made by a model. Commonly known measures are mean absolute error and root mean squared error for regression, and accuracy and area under the ROC curve (AUC) for classification.

Evaluators, like Inducers, should be functions with an S3 class for nicer printing. Most evaluators only need the result made by predict(), but others could also need to access other information, like the original dataset, or the trained model. They should therefore be functions of the form

function(object, task, learner, ...)

Like Inducers, Evaluators should be available through an evl environment:

mae <- evl$mae

identical(mae, EvaluatorMAE)
#> [1] TRUE

mae
#> Evaluator: Mean Absolute Error
#> Configuration: ()

mae(prediction)
#> [1] 6.5