dasf.ml.xgboost
Init module for XGBoost ML algorithms.
Submodules
Classes
Class representing a Fit operation of the pipeline. |
Package Contents
- class dasf.ml.xgboost.XGBRegressor(max_depth=None, max_leaves=None, max_bin=None, grow_policy=None, learning_rate=None, n_estimators=100, verbosity=None, objective=None, booster=None, tree_method=None, n_jobs=None, gamma=None, min_child_weight=None, max_delta_step=None, subsample=None, sampling_method=None, colsample_bytree=None, colsample_bylevel=None, colsample_bynode=None, reg_alpha=None, reg_lambda=None, scale_pos_weight=None, base_score=None, random_state=None, num_parallel_tree=None, monotone_constraints=None, interaction_constraints=None, importance_type=None, gpu_id=None, validate_parameters=None, predictor=None, enable_categorical=False, max_cat_to_onehot=None, eval_metric=None, early_stopping_rounds=None, callbacks=None, **kwargs)[source]
Bases:
dasf.transforms.Fit
,dasf.transforms.FitPredict
,dasf.transforms.Predict
Class representing a Fit operation of the pipeline.
- __xgb_cpu
- __xgb_mcpu
- _lazy_fit_cpu(X, y=None, sample_weight=None, *args, **kwargs)[source]
Respective lazy fit mocked function for CPUs.
- _lazy_fit_gpu(X, y=None, sample_weight=None, *args, **kwargs)[source]
Respective lazy fit mocked function for GPUs.
- _fit_cpu(X, y=None, sample_weight=None)[source]
Respective immediate fit mocked function for local CPU(s).
- _fit_gpu(X, y=None, sample_weight=None, *args, **kwargs)[source]
Respective immediate fit mocked function for local GPU(s).
- _lazy_predict_cpu(X, sample_weight=None, **kwargs)[source]
Respective lazy predict mocked function for CPUs.
- _lazy_predict_gpu(X, sample_weight=None, **kwargs)[source]
Respective lazy predict mocked function for GPUs.