diff --git a/lazypredict/Supervised.py b/lazypredict/Supervised.py index 70cd835..af2780d 100644 --- a/lazypredict/Supervised.py +++ b/lazypredict/Supervised.py @@ -1,14 +1,9 @@ -""" -Supervised Models -""" - import numpy as np import pandas as pd from tqdm import tqdm -import datetime import time from sklearn.pipeline import Pipeline -from sklearn.impute import SimpleImputer, MissingIndicator +from sklearn.impute import SimpleImputer from sklearn.preprocessing import StandardScaler, OneHotEncoder, OrdinalEncoder from sklearn.compose import ColumnTransformer from sklearn.utils import all_estimators @@ -24,8 +19,6 @@ ) import warnings import xgboost - -# import catboost import lightgbm warnings.filterwarnings("ignore")