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
-
Modifier and type Method Description static CatBoostModel loadModel Load the CatBoost model.
int 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.
int getTreeCount Return the number of trees in the model.
int getUsedCategoricFeatureCount Return the number of categorical features in the model.
int getUsedNumericFeatureCount Return the number of numerical features in the model.
CatBoostPredictions or void predict Several methods for applying a model to different types of objects and input features.
void close Implements the
Close
method of thejava.lang.AutoCloseable
interface. Ensures the support of the try-with-resources Java statement.Modifier and type Method Description static CatBoostModel loadModel Load the CatBoost model.
int 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.
int getTreeCount Return the number of trees in the model.
int getUsedCategoricFeatureCount Return the number of categorical features in the model.
int getUsedNumericFeatureCount Return the number of numerical features in the model.
CatBoostPredictions or void predict Several methods for applying a model to different types of objects and input features.
void close Implements the
Close
method of thejava.lang.AutoCloseable
interface. Ensures the support of the try-with-resources Java statement.