Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!
Get Started for Free

predict.CoreModel

Prediction using constructed model


Description

Using a previously built model and new data, predicts the class value and probabilities for classification problem and function value for regression problem.

Usage

## S3 method for class 'CoreModel'
predict(object, newdata, ..., costMatrix=NULL, 
                            type=c("both","class","probability"))

Arguments

object

The model structure as returned by CoreModel.

newdata

Data frame with fresh data.

costMatrix

Optional cost matrix can provide nonuniform costs for classification problems.

type

Controls what will be return value in case of classification.

...

Other model dependent options for prediction. See helpCore.

Details

The function uses the object structure as returned by CoreModel and applies it on the data frame newdata. The newdata must be transformable using the formula specified for building the model (with dependent variable removed). If the dependent variable is present in newdata, it is ignored.

Optional cost matrix can provide nonuniform costs for classification problems. For regression problem this parameter is ignored. The costs can be different from the ones used for building the model in CoreModel.

Value

For regression model a vector of predicted values for given input instances. For classification problem the parameter type controls what is returned. With default value "both" function returns a list with two components class and probabilities containing predicted class values and probabilities for all class values, respectively. With type set to "class" or "probability" the function returns only the selected component as vector or matrix.

Author(s)

Marko Robnik-Sikonja, Petr Savicky

See Also

Examples

# use iris data set

# build random forests model with certain parameters
modelRF <- CoreModel(Species ~ ., iris, model="rf", 
              selectionEstimator="MDL",minNodeWeightRF=5,rfNoTrees=100)
print(modelRF)

# prediction with node distribution
pred <- predict(modelRF, iris, rfPredictClass=FALSE, type="both")
# print(pred)

destroyModels(modelRF) # clean up

CORElearn

Classification, Regression and Feature Evaluation

v1.56.0
GPL-3
Authors
Marko Robnik-Sikonja and Petr Savicky
Initial release
2021-03-23

We don't support your browser anymore

Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.