nlpatl.models.classification.xgboost_classification

class nlpatl.models.classification.xgboost_classification.XGBoostClassification(model_config={}, name='xgboost_classification')

Bases: nlpatl.models.classification.sklearn_classification.SkLearnClassification

A wrapper of xgboost classification class.

Parameters

• model_config (dict) – Model paramateters. Refer to https://xgboost.readthedocs.io/en/stable/python/python_api.html

• name (str) – Name of this classification

>>> import nlpatl.models.classification as nmcla
>>> model = nmcla.XGBoostClassification()

predict_proba(x, predict_config={})

Parameters

• x (np.ndarray) – Raw features

• predict_config (dict) – Model prediction paramateters. Refer to https://xgboost.readthedocs.io/en/stable/python/python_api.html

Returns

Feature and probabilities

Return type

nlptatl.dataset.Dataset

train(x, y)

Parameters

• x (np.ndarray) – Raw features

• y (list of string, int or float or np.ndarray.) – Label of data inputs