CatBoostModel

Basic information

java.lang.Object
ai.catboost.CatBoostModel

public class CatBoostModel extends java.lang.Object implements java.lang.AutoCloseable

Basic model application methods.

Methods

loadModel

Load the CatBoost model.

Modifier and type: static CatBoostModel

getPredictionDimension

Return the dimension of the model's 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

getTreeCount

Return the number of trees in the model.

Modifier and type: int

getUsedCategoricFeatureCount

Return the number of categorical features in the model.

Modifier and type: int

getUsedNumericFeatureCount

Return the number of numerical features in the model.

Modifier and type: int

predict

Several methods for applying a model to different types of objects and input features.

Modifier and type: CatBoostPredictions or void

close

Implements the Close method of the java.lang.AutoCloseable interface. Ensures the support of the try-with-resources Java statement.

Modifier and type: void