text_plain_output_1.png
import pandas as pd

import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)

nrows = None

train_df = pd.read_csv('../input/tabular-playground-series-mar-2022/train.csv', index_col='row_id', nrows=nrows)

features = [c for c in train_df.columns if c not in ['congestion']]
X = train_df[features]
y = train_df['congestion']
features