=[g dZddlmZddlmZddlmZmZGddeZGddeZ Gd d eZ Gd d eZ Gd deZ GddeZ y)z1Xgboost pyspark integration submodule for params.)Dict)TypeConverters)ParamParamscNeZdZUdZeej ddZded<y)HasArbitraryParamsDictz This is a Params based class that is extended by _SparkXGBParams and holds the variable to store the **kwargs parts of the XGBoost input. arbitrary_params_dictzarbitrary_params_dict This parameter holds all of the additional parameters which are not exposed as the XGBoost Spark estimator params but can be recognized by underlying XGBoost library. It is stored as a dictionary.z Param[Dict]N) __name__ __module__ __qualname____doc__rr_dummyr __annotations__O/var/www/html/bid-api/venv/lib/python3.12/site-packages/xgboost/spark/params.pyrr s/ ,1  D,=rrcZeZdZdZeej ddejZ y)HasBaseMarginColz This is a Params based class that is extended by _SparkXGBParams and holds the variable to store the base margin column part of XGboost. base_margin_colz6This stores the name for the column of the base margin typeConverterN) r r r r rrrrtoStringrrrrrrs-  @$-- OrrcneZdZdZeej ddejZ dfd Z xZ S)HasFeaturesColsz Mixin for param features_cols: a list of feature column names. This parameter is taken effect only when use_gpu is enabled. features_colszfeature column names.rcFt||jgy)N)rsuper__init__ _setDefaultself __class__s rrzHasFeaturesCols.__init__5s  r*rreturnN) r r r r rrrr toListStringrr __classcell__r#s@rrr(s8  $11 M++rrcneZdZdZeej ddejZ dfd Z xZ S)HasEnableSparseDataOptimz This is a Params based class that is extended by _SparkXGBParams and holds the variable to store the boolean config of enabling sparse data optimization. enable_sparse_data_optima4This stores the boolean config of enabling sparse data optimization, if enabled, Xgboost DMatrix object will be constructed from sparse matrix instead of dense matrix. This config is disabled by default. If most of examples in your training dataset contains sparse features, we suggest to enable this config.rcFt||jdy)NF)r+rr!s rrz!HasEnableSparseDataOptim.__init__Js  %8rr$) r r r r rrrr toBooleanr+rr'r(s@rr*r*:s> % " W%.. 99rr*cZeZdZdZeej ddejZ y) HasQueryIdColz8 Mixin for param qid_col: query id column name. qid_colzquery id column namerN) r r r r rrrrrr0rrrr/r/Os- $-- Grr/cfeZdZUdZeej ddejZ de d<y)HasContribPredictionColz Mixin for param pred_contrib_col: contribution prediction column name. Output is a 3-dim array, with (rows, groups, columns + 1) for classification case. Else, it can be a 2 dimension for regression case. pred_contrib_colz0feature contributions to individual predictions.rz Param[str]N) r r r r rrrrrr3rrrrr2r2\s5&+ :$-- &lrr2N)r typingrpyspark.ml.paramrpyspark.ml.param.sharedrrrrrr*r/r2rrrr7s[7,1 V  v +f+$9v9* F  f r