CatBoost for PySpark API Reference

catboost_spark Package

Classes

PoolLoadParams([delimiter, hasHeader])

Parameters

QuantizationParams([borderCount, …])

Parameters

Pool(data_frame_or_java_object[, …])

CatBoost’s abstraction of a dataset.

CatBoostClassificationModel(java_model)

Classification model trained by CatBoost.

CatBoostClassifier([allowConstLabel, …])

Class to train CatBoostClassificationModel

CatBoostRegressionModel(java_model)

Regression model trained by CatBoost.

CatBoostRegressor([allowConstLabel, …])

Class to train CatBoostRegressionModel

EAutoClassWeightsType(value)

An enumeration.

EBootstrapType(value)

An enumeration.

EBorderSelectionType(value)

An enumeration.

ECalcTypeShapValues(value)

An enumeration.

EExplainableModelOutput(value)

An enumeration.

EFstrType(value)

An enumeration.

ELeavesEstimation(value)

An enumeration.

ELeavesEstimationStepBacktracking(value)

An enumeration.

ELoggingLevel(value)

An enumeration.

EModelShrinkMode(value)

An enumeration.

EModelType(value)

An enumeration.

ENanMode(value)

An enumeration.

EOverfittingDetectorType(value)

An enumeration.

EPreCalcShapValues(value)

An enumeration.

ESamplingFrequency(value)

An enumeration.

ESamplingUnit(value)

An enumeration.

EScoreFunction(value)

An enumeration.

Indices and tables