Data Engineering and BI courses are free this week!
predict.FFTrees predicts binary classification outcomes or their probabilities from newdata
for an FFTrees object.
# S3 method for FFTrees
predict(
object = NULL,
newdata = NULL,
tree = 1,
type = "class",
sens.w = NULL,
method = "laplace",
data = NULL,
...
)Either a logical vector of predictions, or a matrix of class probabilities.
An FFTrees object created by the FFTrees function.
dataframe. A data frame of test data.
integer. Which tree in the object should be used? By default, tree = 1 is used.
string. What should be predicted? Can be "class", which returns a vector of class predictions, "prob" which returns a matrix of class probabilities,
or "both" which returns a matrix with both class and probability predictions.
deprecated
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.
print.FFTrees for printing FFTs;
plot.FFTrees for plotting FFTs;
summary.FFTrees for summarizing FFTs;
FFTrees for creating FFTs from and applying them to data.
# 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 object from the training data:
breast.fft <- FFTrees(
formula = diagnosis ~ .,
data = breast.train
)
# Predict classification outcomes for test data:
breast.fft.pred <- predict(breast.fft,
newdata = breast.test
)
# Predict class probabilities for test data:
breast.fft.pred <- predict(breast.fft,
newdata = breast.test,
type = "prob"
)
Run the code above in your browser using DataLab