diff --git a/ml_garden/implementation/tabular/autogluon/model.py b/ml_garden/implementation/tabular/autogluon/model.py index c072fb2..bb58450 100644 --- a/ml_garden/implementation/tabular/autogluon/model.py +++ b/ml_garden/implementation/tabular/autogluon/model.py @@ -16,9 +16,6 @@ def __init__(self, **params): self.model: TabularPredictor = None def fit(self, X: pd.DataFrame, y: pd.Series, eval_set=None, verbose=True) -> None: - - # Autogluon SHOULD ignore encoders and previous steps - train_data = X.copy() train_data["label"] = y if eval_set is not None and len(eval_set) > 0: