org.apache.ignite.ml.catboost

Class CatboostRegressionModel

java.lang.Object

org.apache.ignite.ml.catboost.CatboostRegressionModel

All Implemented Interfaces:

AutoCloseable, Model<NamedVector,Double>

public class CatboostRegressionModel
extends Object
implements Model<NamedVector,Double>

CatboostRegressionModel imported and wrapped to be compatible with Apache Ignite infrastructure.

Constructor Summary

Constructors

Constructor and Description
CatboostRegressionModel(ai.catboost.CatBoostModel model) Constructs a new instance of Catboost model wrapper.

Method Summary

Modifier and Type	Method and Description
void	close()
Double	predict(NamedVector input) Make a prediction for the specified input arguments.

Methods inherited from class java.lang.Object

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

Constructor Detail

CatboostRegressionModel

public CatboostRegressionModel(ai.catboost.CatBoostModel model)

Constructs a new instance of Catboost model wrapper.

Parameters:

model - Catboost Model

Method Detail

predict

public Double predict(NamedVector input)

Make a prediction for the specified input arguments.

Specified by:

predict in interface Model<NamedVector,Double>

Parameters:

input - Input arguments.

Returns:

Prediction result.

close

public void close()

Specified by:

close in interface AutoCloseable

Specified by:

close in interface Model<NamedVector,Double>