diff --git a/src/preprocessing.py b/src/preprocessing.py index 37aa537..6849acf 100644 --- a/src/preprocessing.py +++ b/src/preprocessing.py @@ -23,6 +23,7 @@ def drop_null_values(df_list): for df in df_list: df.drop(columns="tipo_marchas", inplace=True) df.drop(columns="descuento", inplace=True) + df.drop(columns="ciudad", inplace=True) df.dropna(inplace=True) return df_list @@ -32,7 +33,6 @@ def encode_columns(df_list): files = [ "ao", "asientos", - "ciudad", "combustible", "consumo", "kilometros",