Add EqualizedOddsImprovement

Author: fealho

sdmetrics/single_table/__init__.py

@@ -77,6 +77,7 @@
 from sdmetrics.single_table.privacy.numerical_sklearn import NumericalLR, NumericalMLP, NumericalSVR
 from sdmetrics.single_table.privacy.radius_nearest_neighbor import NumericalRadiusNearestNeighbor
 from sdmetrics.single_table.table_structure import TableStructure
+from sdmetrics.single_table.equalized_odds import EqualizedOddsImprovement
 
 __all__ = [
     'bayesian_network',
@@ -140,4 +141,5 @@
     'TableStructure',
     'DCRBaselineProtection',
     'DCROverfittingProtection',
+    'EqualizedOddsImprovement',
 ]

sdmetrics/single_table/data_augmentation/utils.py

@@ -3,31 +3,11 @@
 import pandas as pd
 
 from sdmetrics._utils_metadata import _process_data_with_metadata, _validate_single_table_metadata
-
-
-def _validate_tables(real_training_data, synthetic_data, real_validation_data):
-    """Validate the tables of the Data Augmentation metrics."""
-    tables = [real_training_data, synthetic_data, real_validation_data]
-    if any(not isinstance(table, pd.DataFrame) for table in tables):
-        raise ValueError(
-            '`real_training_data`, `synthetic_data` and `real_validation_data` must be '
-            'pandas DataFrames.'
-        )
-
-
-def _validate_prediction_column_name(prediction_column_name):
-    """Validate the prediction column name of the Data Augmentation metrics."""
-    if not isinstance(prediction_column_name, str):
-        raise TypeError('`prediction_column_name` must be a string.')
-
-
-def _validate_classifier(classifier):
-    """Validate the classifier of the Data Augmentation metrics."""
-    if classifier is not None and not isinstance(classifier, str):
-        raise TypeError('`classifier` must be a string or None.')
-
-    if classifier != 'XGBoost':
-        raise ValueError('Currently only `XGBoost` is supported as classifier.')
+from sdmetrics.single_table.utils import (
+    _validate_classifier,
+    _validate_prediction_column_name,
+    _validate_tables,
+)
 
 
 def _validate_fixed_recall_value(fixed_recall_value):