diff --git a/notebooks/bias_reduction.ipynb b/notebooks/bias_reduction.ipynb index fa35d2b..2e71d6d 100644 --- a/notebooks/bias_reduction.ipynb +++ b/notebooks/bias_reduction.ipynb @@ -159,7 +159,7 @@ " scatter_schema = ['model_uuid', performance_col, 'attribute_name', 'attribute_value', bias_metric, disparity_metric, 'model_tag']\n", " if flip_disparity:\n", " disparity_df[disparity_metric]= disparity_df.apply(lambda x: 1/x[disparity_metric] , axis=1)\n", - " scatter = pd.merge(evals_df, disparity_df, how='left', on=['model_uuid'], left_index=True, sort=True, copy=True)\n", + " scatter = pd.merge(evals_df, disparity_df, how='left', on=['model_uuid'], sort=True, copy=True)\n", " scatter = scatter[['model_uuid', performance_col, 'attribute_name', 'attribute_value', bias_metric, disparity_metric]].copy()\n", " scatter['model_tag'] = 'Other Models'\n", " scatter.sort_values('model_precision', ascending = False, inplace=True, ignore_index=True)\n",