CatBoostPredictions

Basic information

java.lang.Object
    ai.catboost.CatBoostPredictions

public class CatBoostPredictions extends java.lang.ObjectCatBoost model prediction.

A wrapper that provides methods for making convenient predictions for certain classes.

Constructor

CatBoostPredictions(int objectCount,
                    int predictionDimension)

Construct the CatBoost model prediction based on the number of objects and the model prediction dimension.

Methods

copyObjectPredictions

Copy the prediction for the specified object.

Modifier and type: double or void

copyRowMajorPredictions

Return a row-major copy of the prediction matrix.

Modifier and type: double

get

Return the model prediction for a particular object in a particular dimension.

Modifier and type: double

getObjectCount

Return the number of objects that contain predictions in the current instance of the class.

Modifier and type: int

getPredictionDimension

Return the dimension of the prediction. The value corresponds to the dimension of the vector of predictions. For example, it is equal to N for multiclassification tasks in case of N classes.

Modifier and type: int