aibox.nlp.estimators.regression.xgboost_regressor

Regressor XGBoost.

Classes

XGBoostRegressor([n_estimators, ...])

Regressor XGBoost.
class aibox.nlp.estimators.regression.xgboost_regressor.XGBoostRegressor(n_estimators: int = 100, learning_rate: float | None = None, grow_policy: int | None = None, booster: str | None = None, tree_method: str | None = None, random_state: int | None = None)[código-fonte]

Base: Estimator

Regressor XGBoost. Essa classe é um wrapper de xgboost.XGBoostRegressor.

Parâmetros:

  • n_estimators (int)

  • learning_rate (float)

  • grow_policy (int)

  • booster (str)

  • tree_method (str)

  • random_state (int | None)

predict(X: list[int] | list[float] | ndarray[int32] | ndarray[float32] | Tensor, **kwargs) → ndarray[código-fonte]

Realiza a predição utilizando os parâmetros atuais do modelo.

Parâmetros:

  • X (list[int] | list[float] | ndarray[int32] | ndarray[float32] | Tensor) – dados de entrada com shape (n_samples, n_features).

  • **kwargs – parâmetros extras que podem ser passados para alguns estimadores.

Retorna:

predições com shape (n_samples,).

Tipo de retorno:

ndarray

fit(X: list[int] | list[float] | ndarray[int32] | ndarray[float32] | Tensor, y: list[int] | list[float] | ndarray[int32] | ndarray[float32] | Tensor, **kwargs)[código-fonte]

Realiza o treinamento do estimador.

Parâmetros:
property hyperparameters: dict

Hiper-parâmetros do modelo. Inclui a seed randômica. Estrutura do dicionário varia entre diferentes estimadores.

Retorna:

dicionário de hiper-parâmetros.

property params: dict

Retorna um dicionário com os parâmetros para esse estimador.

Os parâmetros retornados descrevem totalmente o estado do modelo (e,g. pesos de uma rede, superfícies de decisão, estrutura da árvore de decisão, etc).

Retorna:

parâmetros do estimador.

property random_state: int

Seed randômica utilizada pelo estimador.

Retorna:

seed.