rxPredict.rxNaiveBayes: Prediction for Large Data Naive Bayes Classifiers

  • [アーティクル]

Description

Calculate predicted or fitted values for a data set from an rxNaiveBayes object.

Usage

 ## S3 method for class `rxNaiveBayes':
rxPredict  (modelObject, data = NULL, outData = NULL, type = c("class", "prob"), prior = NULL,
      predVarNames = NULL, writeModelVars = FALSE, extraVarsToWrite = NULL, checkFactorLevels = TRUE,
        ...  )

Arguments

modelObject

object returned from a call to rxNaiveBayes.

data

either a data source object, a character string specifying a .xdf file, or a data frame object.

outData

file or existing data frame to store predictions; can be same as the input file or NULL. If not NULL, must be an .xdf file if data is an .xdf file or a data frame if data is a data frame.

type

character string specifying the type of predicted values to be returned. Supported choices are

  • "class" - a vector of predicted classes.
  • "prob" - a matrix of predicted class probabilities whose columns are the probability of the first, second, etc. class.

prior

a vector of prior probabilities. If unspecified, the class proportions of the data counts in the training set are used. If present, they should be specified in the order of the factor levels of the response and they must be all non-negative and sum to 1.

predVarNames

character vector specifying name(s) to give to the prediction results.

writeModelVars

logical value. If TRUE, and the output file is different from the input file, variables in the model will be written to the output file in addition to the predictions. If variables from the input data set are transformed in the model, the transformed variables will also be written out.

extraVarsToWrite

NULL or character vector of additional variables names from the input data to include in the outData. If writeModelVars is TRUE, model variables will be included as well.

checkFactorLevels

logical value. If TRUE, the factor levels for the data will be verified against factor levels in the model. Setting to FALSE can speed up computations if using lots of factors.

...

additional arguments to be passed directly to rxDataStep such as removeMissingsOnRead, overwrite, blocksPerRead, reportProgress, xdfCompressionLevel.

Details

Prediction for large data models requires both a fitted model object and a data set, either the original data (to obtain fitted values and residuals) or a new data set containing the same set of variables as the original fitted model. Notice that this is different from the behavior of predict, which can usually work on the original data simply by referencing the fitted model.

Value

Depending on the form of data, this function variously returns a data frame or a data source representing a .xdf file.

Author(s)

Microsoft Corporation Microsoft Technical Support

References

Naive Bayes classifier https://en.wikipedia.org/wiki/Naive_Bayes_classifier .

See Also

rxNaiveBayes.

Examples


 # multi-class classification with a data.frame
 iris.nb <- rxNaiveBayes(Species ~ ., data = iris)
 iris.nb

 # prediction
 iris.nb.pred <- rxPredict(iris.nb, iris)
 iris.nb.pred
 table(iris.nb.pred[["Species_Pred"]], iris[["Species"]])