get_cutoffs: Get cutoff values for group predictions

Description Usage Arguments Value Examples

View source: R/predict.R

Description

Get cutoff values for group predictions

Usage

1

Arguments

x

Data frame from predict.model_list where outcome_groups or risk_groups was specified

Value

A message is printed about the thresholds. If outcome_groups were defined the return value is a single numeric value, the threshold used to separate predicted probabilities into outcome groups. If risk_groups were defined the return value is a data frame with one column giving the group names and another column giving the minimum predicted probability for an observation to be in that group.

Examples

1
2
3
4
machine_learn(pima_diabetes[1:20, ], patient_id, outcome = diabetes,
              models = "xgb", tune = FALSE) %>%
  predict(risk_groups = 5) %>%
  get_cutoffs()

healthcareai documentation built on Sept. 2, 2018, 1:03 a.m.