FFTrees (version 1.3.5)

predict.FFTrees: Predict new data from an FFTrees x

Description

Predict new data from an FFTrees x

Usage

# S3 method for FFTrees
predict(object = NULL, newdata = NULL, data = NULL,
  tree = 1, type = "class", sens.w = NULL, method = "laplace", ...)

Arguments

object

An FFTrees object created from the FFTrees() function.

newdata

dataframe. A dataframe of test data

tree

integer. Which tree in the object should be used?

type

string. What should be predicted? Can be "class", which returns a vector of class predictions, or "prob" which returns a matrix of class probabilities.

sens.w, data

depricated

method

string. Method of calculating class probabilities. Either 'laplace', which applies the Laplace correction, or 'raw' which applies no correction.

...

Additional arguments passed on to predict()

Value

A logical vector of predictions

Examples

Run this code
# NOT RUN {

  # Create training and test data

  set.seed(100)
  breastcancer <- breastcancer[sample(nrow(breastcancer)),]
  breast.train <- breastcancer[1:150,]
  breast.test <- breastcancer[151:303,]

  # Create an FFTrees x from the training data

  breast.fft <- FFTrees(formula = diagnosis ~.,
                              data = breast.train)

 # Predict results for test data
  breast.fft.pred <- predict(breast.fft,
                             data = breast.test)

# }

Run the code above in your browser using DataLab