convert sdmetrics quant metrics to classic key:value pair

Author: sniperyyc

netshare/pre_post_processors/netshare/choose_best_model.py

@@ -4,7 +4,7 @@
 import pandas as pd
 import numpy as np
 
-from .util import create_sdmetrics_config
+from .util import create_sdmetrics_config, convert_sdmetricsConfigQuant_to_fieldValueDict
 from sdmetrics.reports.timeseries import QualityReport
 
 
@@ -18,7 +18,9 @@ def compare_rawdf_syndfs(
         comparison_type='quantitative')
     report = QualityReport(config_dict=sdmetrics_config['config'])
     report.generate(raw_df, syn_dfs[0], sdmetrics_config['metadata'])
-    print("\n\n\n", report.dict_metric_scores)
+    metricValueDict = convert_sdmetricsConfigQuant_to_fieldValueDict(
+        report.dict_metric_scores)
+    print(metricValueDict)
 
     ()+1
 

netshare/pre_post_processors/netshare/util.py

@@ -3,6 +3,7 @@
 import pickle
 import math
 import json
+import ast
 import socket
 import struct
 import ipaddress
@@ -25,6 +26,26 @@
 from ...model_managers.netshare_manager.netshare_util import get_configid_from_kv
 
 
+def convert_sdmetricsConfigQuant_to_fieldValueDict(
+    sdmetricsConfigQuant
+):
+    '''Convert the sdmetricsConfigQuant to fieldValueDict
+    Args:
+        sdmetricsConfigQuant (dict): returned by create_sdmetrics_config(...,  comparison_type='quantitative')
+    Returns:
+        fieldValueDict (dict): {field_name: value}
+    '''
+
+    fieldValueDict = {}
+    for metric_type, metrics in sdmetricsConfigQuant.items():
+        for metric_class_name, metric_class in metrics.items():
+            for field_name, field_value in metric_class.items():
+                fieldValueDict[ast.literal_eval(
+                    field_name)[0]] = field_value[0][0]
+
+    return fieldValueDict
+
+
 def create_sdmetrics_config(
     config_pre_post_processor,
     comparison_type='both'