text_plain_output_1.png
X_train, y_train = (train_df.drop('quality', axis=1), train_df['quality'])
X_val, y_val = (val_df.drop('quality', axis=1), val_df['quality'])
X_test, y_test = (test_df.drop('quality', axis=1), test_df['quality'])
(X_train.shape, y_train.shape, X_val.shape, y_val.shape, X_test.shape, y_test.shape)