Source code for autogluon.tabular.models.xt.xt_model


from autogluon.core.constants import REGRESSION, QUANTILE

from ..rf.rf_model import RFModel


[docs]class XTModel(RFModel): """ Extra Trees model (scikit-learn): https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.ExtraTreesClassifier.html#sklearn.ensemble.ExtraTreesClassifier """ def _get_model_type(self): if self.problem_type == REGRESSION: from sklearn.ensemble import ExtraTreesRegressor return ExtraTreesRegressor elif self.problem_type == QUANTILE: from ..rf.rf_quantile import ExtraTreesQuantileRegressor return ExtraTreesQuantileRegressor else: from sklearn.ensemble import ExtraTreesClassifier return ExtraTreesClassifier