Skip to content

Commit 5b449e7

Browse files
committed
add metric_to_fix in parameters
1 parent fbc5d37 commit 5b449e7

File tree

1 file changed

+2
-1
lines changed
  • sdmetrics/single_table/data_augmentation

1 file changed

+2
-1
lines changed

sdmetrics/single_table/data_augmentation/base.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -207,6 +207,7 @@ def compute_breakdown(
207207
fixed_recall_value,
208208
cls.metric_name,
209209
)
210+
metric_to_fix = 'recall' if cls.metric_name == 'precision' else 'precision'
210211
result = {
211212
'real_data_baseline': trainer.get_scores(
212213
preprocessed_tables['real_training_data'],
@@ -223,7 +224,7 @@ def compute_breakdown(
223224
'prediction_column_name': trainer.prediction_column_name,
224225
'minority_class_label': trainer.minority_class_label,
225226
'classifier': trainer._classifier_name,
226-
'fixed_recall_value': trainer.fixed_value,
227+
f'fixed_{metric_to_fix}_value': trainer.fixed_value,
227228
},
228229
}
229230
result['score'] = max(

0 commit comments

Comments
 (0)