application_vnd.jupyter.stderr_output_2.png application_vnd.jupyter.stderr_output_1.png
import pandas as pd

df_train = pd.read_csv('/kaggle/input/aqueous-solubility-predictioin/train.csv')
df_test = pd.read_csv('/kaggle/input/aqueous-solubility-predictioin/test.csv')

df_train.shape

df_train.columns

X = df_train.drop(columns=['ID', 'Name', 'InChI', 'InChIKey', 'SMILES', 'SD', 'Ocurrences', 'Group', 'comp_id'])

reg = setup(X, target='Solubility', session_id=123, train_size=0.8)