from autogluon.tabular import TabularDataset, TabularPredictor

# Load the dataset
train_df = TabularDataset('data/train.csv')