Packages

•  package root

  Definition Classes

  root

•  package ai

  Definition Classes

  root

•  package catboost

  Definition Classes

  ai

•  package spark

  CatBoost is a machine learning algorithm that uses gradient boosting on decision trees.

  CatBoost is a machine learning algorithm that uses gradient boosting on decision trees.

  Overview

  This package provides classes that implement interfaces from Apache Spark Machine Learning Library (MLLib).

  For binary and multi- classification problems use CatBoostClassifier, for regression use CatBoostRegressor.

  These classes implement usual fit method of org.apache.spark.ml.Predictor that accept a single org.apache.spark.sql.DataFrame for training, but you can also use other fit method that accepts additional datasets for computing evaluation metrics and overfitting detection similarily to CatBoost's other APIs.

  This package also contains Pool class that is CatBoost's abstraction of a dataset. It contains additional information compared to simple org.apache.spark.sql.DataFrame.

  It is also possible to create Pool with quantized features before training by calling quantize method. This is useful if this dataset is used for training multiple times and quantization parameters do not change. Pre-quantized Pool allows to cache quantized features data and so do not re-run feature quantization step at the start of an each training.

  Detailed documentation is available on https://catboost.ai/docs/

  Definition Classes

  catboost

•  package impl

  Definition Classes

  spark

•  package pyspark_wrapper_generator
• CtrFeatures
• CtrsContext
• ExpressionEncoderSerializer
• FeatureImportanceCalcer
•  package params

  Definition Classes

  spark

ai.catboost.spark

impl 