Documentation Center |
labels = predict(ens,X)
[labels,score]
= predict(ens,X)
[labels,...] = predict(ens,X,Name,Value)
labels = predict(ens,X) returns a vector of predicted class labels for a matrix X, based on ens, a trained full or compact classification ensemble.
[labels,score] = predict(ens,X) also returns scores for all classes.
[labels,...] = predict(ens,X,Name,Value) predicts classifications with additional options specified by one or more Name,Value pair arguments.
ens |
A classification ensemble created by fitensemble, or a compact classification ensemble created by compact. |
X |
A matrix where each row represents an observation, and each column represents a predictor. The number of columns in X must equal the number of predictors in ens. |
Specify optional comma-separated pairs of Name,Value arguments. Name is the argument name and Value is the corresponding value. Name must appear inside single quotes (' '). You can specify several name and value pair arguments in any order as Name1,Value1,...,NameN,ValueN.
labels |
Vector of classification labels. labels has the same data type as the labels used in training ens. |
score |
A matrix with one row per observation and one column per class. For each observation and each class, the score generated by each tree is the probability of this observation originating from this class computed as the fraction of observations of this class in a tree leaf. predict averages these scores over all trees in the ensemble. |
For ensembles, a classification score represents the confidence of a classification into a class. The higher the score, the higher the confidence.
Different ensemble algorithms have different definitions for their scores. Furthermore, the range of scores depends on ensemble type. For example:
AdaBoostM1 scores range from –∞ to ∞.
Bag scores range from 0 to 1.
Train a boosting ensemble for the ionosphere data, and predict the classification of the mean of the data:
load ionosphere; ada = fitensemble(X,Y,'AdaBoostM1',100,'tree'); Xbar = mean(X); [ypredict score] = predict(ada,Xbar) ypredict = 'g' score = -2.9460 2.9460