XGBoostClassifier class

Definition

XGBoost Classifier class.

XGBoostClassifier(random_state=0, n_jobs=1, problem_info=None, **kwargs)
Inheritance
builtins.object
sklearn.base.BaseEstimator
XGBoostClassifier
builtins.object
sklearn.base.ClassifierMixin
XGBoostClassifier
azureml.automl.runtime.shared.model_wrappers._AbstractModelWrapper
XGBoostClassifier

Parameters

random_state
int or RandomState

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

Number of parallel threads.

kwargs

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

Methods

fit(X, y, **kwargs)

Fit function for XGBoost Classifier model.

get_model()

Return XGBoost Classifier model.

Else returns None.

get_params(deep=True)

Return parameters for XGBoost Classifier model.

predict(X)

Prediction function for XGBoost Classifier model.

predict_proba(X)

Prediction class probabilities for X for XGBoost Classifier model.

fit(X, y, **kwargs)

Fit function for XGBoost Classifier model.

fit(X, y, **kwargs)

Parameters

X
ndarray

Input data.

y
ndarray

Input target values.

kwargs

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(deep=True)

Return parameters for XGBoost Classifier model.

get_params(deep=True)

Parameters

deep
boolean
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(X)

Prediction function for XGBoost Classifier model.

predict(X)

Parameters

X
ndarray

Input data.

Returns

Prediction values from XGBoost Classifier model.

predict_proba(X)

Prediction class probabilities for X for XGBoost Classifier model.

predict_proba(X)

Parameters

X
ndarray

Input data.

Returns

Prediction probability values from XGBoost Classifier model.