/auprc

Package for calculating AUPRC (Area Under Precision-Recall Curve) in R

Primary LanguageROtherNOASSERTION

auprc

You can install this package using

devtools::install_github("DominikRafacz/auprc")

Two main functionalities of this package are calculating plotting the precision-recall curve and calculating area under it.

auprc(prob, y_truth, positive_value)
precision_recall_curve(prob, y_truth, positive_value)

This package registers measure for mlr3 framework. You can access it by using

msr("classif.auprc")

or

mlr_measures$get("classif.auprc")

Important! If you want to make sure that AUPRC for mlr3 is computed correctly, you have to specify parameter positive during creation of new ClassifTask.