myogestic.models.definitions.catboost_models.predict#
- myogestic.models.definitions.catboost_models.predict(model, input, is_classifier)[source]#
Predict with a CatBoost model.
- Parameters:
model (_CatBoostBase) – The CatBoost model to predict with.
input (np.ndarray) – The input data to predict. The shape of the input data will be (1, n_features, n_samples).
is_classifier (bool) – If the model is a classifier.
- Returns:
The prediction. If the model is a classifier, the prediction will be a np.array. If the model is a regressor, the prediction will be a list of floats.
- Return type: