Package

ai.catboost.spark

params

Permalink

package params

Visibility
  1. Public
  2. All

Type Members

  1. trait ClassifierTrainingParamsTrait extends TrainingParamsTrait

    Permalink

    Params for training CatBoostClassifier.

    Params for training CatBoostClassifier. See documentation on https://catboost.ai/docs/ for details.

  2. trait DatasetParamsTrait extends HasLabelCol with HasFeaturesCol with HasWeightCol

    Permalink

    Common dataset columns parameters

  3. class DurationParam extends Param[Duration]

    Permalink
  4. class EnumParam[E <: Enum[E]] extends Param[E]

    Permalink

    A separate class is needed because default Param[E] won't jsonEncode/Decode properly

  5. trait IgnoredFeaturesParams extends Params

    Permalink
  6. class OrderedStringMapParam[V] extends Param[LinkedHashMap[String, V]]

    Permalink

    supported V types are String, Long, Double or Boolean

  7. class PoolLoadParams extends Params

    Permalink

    Params for loading datasets in CatBoost dsv format

  8. class QuantizationParams extends QuantizationParamsTrait

    Permalink

    Parameters for quantization as a class.

    Parameters for quantization as a class. Used in Pool!.quantize(QuantizationParamsTrait). See documentation on https://catboost.ai/docs/ for details.

  9. trait QuantizationParamsTrait extends Params with IgnoredFeaturesParams with ThreadCountParams

    Permalink

    Parameters for quantization.

    Parameters for quantization. See documentation on https://catboost.ai/docs/ for details.

  10. trait RegressorTrainingParamsTrait extends TrainingParamsTrait

    Permalink

    Params for training CatBoostRegressor.

    Params for training CatBoostRegressor. See documentation at https://catboost.ai/docs/ for details.

  11. trait ThreadCountParams extends Params

    Permalink
  12. trait TrainingParamsTrait extends QuantizationParamsTrait with HasLabelCol with HasFeaturesCol with HasWeightCol

    Permalink

    Params for training CatBoost.

    Params for training CatBoost. See documentation on https://catboost.ai/docs/ for details.

Value Members

  1. object QuantizationParams extends Serializable

    Permalink

Ungrouped