Learn R Programming

healthcareai (version 1.2.4)

calculatePerformance: Generate performance metrics after model has been trained

Description

Generates AU_ROC and AU_PR (including 95

Usage

calculatePerformance(predictions, ytest, type)

Arguments

predictions

A vector of predictions from a machine learning model.

ytest

A vector of the true labels. Must be the same length as predictions.

type

A string. Indicates model type and can be "regression" or "classification". Defaults to SS.

Value

Curves (if classification); otherwise nothing. Prints results.

References

http://healthcareai-r.readthedocs.io

See Also

healthcareai