Views
No views yet
1from datasets import load_dataset
2import imodels
3import numpy as np
4from sklearn.model_selection import GridSearchCV
5import joblib
6
7dataset = load_dataset("imodels/compas-recidivism")
8df = pd.DataFrame(dataset['train'])
9X_train = df.drop(columns=['is_recid'])
10y_train = df['is_recid'].values
11
12df_test = pd.DataFrame(dataset['test'])
13X_test = df.drop(columns=['is_recid'])
14y_test = df['is_recid'].values1from huggingface_hub import hf_hub_url, cached_download
2import joblib
3import pandas as pd
4
5REPO_ID = "imodels/figs-compas-recidivism"
6FILENAME = "sklearn_model.joblib"
7
8model = joblib.load(cached_download(
9 hf_hub_url(REPO_ID, FILENAME)
10))
11
12# model is a `imodels.FIGSClassifier`preds = model.predict(X_test)
print('accuracy', np.mean(preds==y_test))
# accuracy 0.6759165485112416