dasf.ml.xgboost

Init module for XGBoost ML algorithms.

Submodules

Classes

XGBRegressor

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.

_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.

_predict_cpu(X, sample_weight=None, **kwargs)[source]

Respective immediate predict mocked function for local CPU(s).

_predict_gpu(X, sample_weight=None, **kwargs)[source]

Respective immediate predict mocked function for local GPU(s).