ai.catboost

Class CatBoostPredictions

java.lang.Object

ai.catboost.CatBoostPredictions

public class CatBoostPredictions
extends Object

CatBoost model prediction.

Constructor Summary

Constructors

Constructor and Description
CatBoostPredictions(int objectCount, int predictionDimension) Construct CatBoost model prediction based on number of objects and model prediction dimension.

Method Summary

Modifier and Type	Method and Description
double[]	copyObjectPredictions(int objectIndex) Copy object prediction to a separate array.
void	copyObjectPredictions(int objectIndex, double[] predictions) Copy object prediction to a specified array.
double[]	copyRowMajorPredictions() Return row-major copy of prediction matrix.
double	get(int objectIndex, int predictionIndex) Get model prediction for particular object and particular dimension.
int	getObjectCount()
int	getPredictionDimension()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

CatBoostPredictions

public CatBoostPredictions(int objectCount,
                           int predictionDimension)

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

Parameters:

objectCount - Object count.

predictionDimension - CatBoost model prediction dimension.

Method Detail

getObjectCount

public int getObjectCount()

Returns:

Number of objects in prediction.

getPredictionDimension

public int getPredictionDimension()

Returns:

Model prediction dimension.

get

public double get(int objectIndex,
                  int predictionIndex)

Get model prediction for particular object and particular dimension.

Parameters:

objectIndex - Object index.

predictionIndex - Prediction dimension index.

Returns:

Model prediction value.

copyObjectPredictions

public void copyObjectPredictions(int objectIndex,
                                  @NotNull
                                  double[] predictions)

Copy object prediction to a specified array.

Parameters:

objectIndex - Object index.

predictions - Array to copy predictions to.

copyObjectPredictions

@NotNull
public double[] copyObjectPredictions(int objectIndex)

Copy object prediction to a separate array.

Parameters:

objectIndex - Object index.

Returns:

Array with object predictions.

See Also:

copyObjectPredictions(int, double[])

copyRowMajorPredictions

@NotNull
public double[] copyRowMajorPredictions()

Return row-major copy of prediction matrix. Prediction for object with index `i` in dimension `j` will be at `i*getPredictionDimension() + j`.

Returns:

Row-major copy of prediction matrix.