Commit 5b449e7 1 parent fbc5d37 commit 5b449e7 Copy full SHA for 5b449e7
File tree 1 file changed +2
-1
lines changed
sdmetrics/single_table/data_augmentation
1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -207,6 +207,7 @@ def compute_breakdown(
207
207
fixed_recall_value ,
208
208
cls .metric_name ,
209
209
)
210
+ metric_to_fix = 'recall' if cls .metric_name == 'precision' else 'precision'
210
211
result = {
211
212
'real_data_baseline' : trainer .get_scores (
212
213
preprocessed_tables ['real_training_data' ],
@@ -223,7 +224,7 @@ def compute_breakdown(
223
224
'prediction_column_name' : trainer .prediction_column_name ,
224
225
'minority_class_label' : trainer .minority_class_label ,
225
226
'classifier' : trainer ._classifier_name ,
226
- 'fixed_recall_value ' : trainer .fixed_value ,
227
+ f'fixed_ { metric_to_fix } _value ' : trainer .fixed_value ,
227
228
},
228
229
}
229
230
result ['score' ] = max (
You can’t perform that action at this time.
0 commit comments