 Load the data  ```python from datasets import load_dataset import imodels import numpy as np from sklearn.model_selection import GridSearchCV import joblib  dataset = load_dataset("imodels/compas-recidivism") df = pd.DataFrame(dataset['train']) X_train = df.drop(columns=['is_recid']) y_train = df['is_recid'].values  df_test = pd.DataFrame(dataset['test']) X_test = df.drop(columns=['is_recid']) y_test = df['is_recid'].values ```  