maksymiuks / auprc

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

About

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

License:Other


Languages

Language:R 100.0%