XGBoostClassifier Class

  • Reference

XGBoost Classifier class.

Inheritance
sklearn.base.BaseEstimator
XGBoostClassifier
sklearn.base.ClassifierMixin
XGBoostClassifier
azureml.training.tabular.models._abstract_model_wrapper._AbstractModelWrapper
XGBoostClassifier

Constructor

XGBoostClassifier(random_state=0, n_jobs=1, problem_info=None, **kwargs)

Parameters

random_state
int or <xref:np.random.RandomState>
default value: 0

RandomState instance or None, optional (default=None) If int, random_state is the seed used by the random number generator; If RandomState instance, random_state is the random number generator; If None, the random number generator is the RandomState instance used by np.random.

n_jobs
int
default value: 1

Number of parallel threads.

kwargs
Required

Other parameters Check https://xgboost.readthedocs.io/en/latest/parameter.html for more parameters.

problem_info
default value: None

Methods

fit

Fit function for XGBoost Classifier model.
get_model

Return XGBoost Classifier model.

Else returns None.
get_params

Return parameters for XGBoost Classifier model.
predict

Prediction function for XGBoost Classifier model.
predict_proba

Prediction class probabilities for X for XGBoost Classifier model.

fit

Fit function for XGBoost Classifier model.

fit(X, y, **kwargs)

Parameters

X
ndarray
Required

Input data.

y
ndarray
Required

Input target values.

kwargs
Required

other parameters Check https://xgboost.readthedocs.io/en/latest/parameter.html for more parameters.

Returns

Self after fitting the model.

get_model

Return XGBoost Classifier model.

Else returns None.

get_model()

Returns

Returns the fitted model if fit method has been called.

get_params

Return parameters for XGBoost Classifier model.

get_params(deep=True)

Parameters

deep
bool
default value: True

If True, will return the parameters for this estimator and contained subobjects that are estimators.

Returns

Parameters for the XGBoost classifier model.

predict

Prediction function for XGBoost Classifier model.

predict(X)

Parameters

X
ndarray
Required

Input data.

Returns

Prediction values from XGBoost Classifier model.

predict_proba

Prediction class probabilities for X for XGBoost Classifier model.

predict_proba(X)

Parameters

X
ndarray
Required

Input data.

Returns

Prediction probability values from XGBoost Classifier model.