predict_class_train: Predict train data outcome.

View source: R/utils_prediction.R

predict_class_trainR Documentation

Predict train data outcome.

Description

a helper function extracting the predictions for the train, cross-validation and test data set from a classification or regression caretx model.

Usage

predict_class_train(class_model)

predict_reg_train(reg_model)

predict_class_cv(class_model)

predict_reg_cv(reg_model)

predict_class_test(class_model, newdata)

predict_reg_test(reg_model, newdata)

Arguments

class_model

a caretx classification model.

reg_model

a caretx regression model.

newdata

a data frame with test data.

Value

a predx object.


PiotrTymoszuk/caretExtra documentation built on Oct. 15, 2023, 10:03 p.m.