model_pr | R Documentation |
model_pr() returns the Precision - Recall curve, which is based on the confusion matrix (Precision and Recall). The graph allows you to study cases of population imbalance.
model_pr(y_train, y_train_prob, y_test, y_test_prob, type = 3)
y_train |
Values of the response variable in training set. |
y_train_prob |
Probabilities of the response variable in training set. |
y_test |
Values of the response variable in test set. |
y_test_prob |
Probabilities of the response variable in test set. |
type |
1 indicates that the PR curve is plotted using Recall on the x-axis and Precision on the y-axis, while 2 indicates that both Recall and Precision are plotted using probabilities on the x-axis. Value 3 generates both graphs (default). |
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.