@@ -1300,17 +1300,15 @@ def calculate_model_statistics(
1300
1300
roc_dict = cls .apply_dataframe_to_json (json_dict [1 ]["data" ], i , roc_df )
1301
1301
for j in range (len (roc_dict )):
1302
1302
json_dict [1 ]["data" ][j ].update (roc_dict [j ])
1303
- if (roc_dict [j ]["dataMap" ]["_KS_" ] == 1 ):
1304
- fitstat_data = {
1305
- "_KS_" : roc_dict [j ]["dataMap" ]["_KS_" ],
1306
- "_KS2_" : roc_dict [j ]["dataMap" ]["_KS2_" ],
1307
- "_C_" : roc_dict [j ]["dataMap" ]["_C_" ],
1308
- "_Gini_" : roc_dict [j ]["dataMap" ]["_Gini_" ],
1309
- "_Gamma_" : roc_dict [j ]["dataMap" ]["_Gamma_" ],
1310
- "_Tau_" : roc_dict [j ]["dataMap" ]["_Tau_" ]
1311
- }
1312
-
1313
- json_dict [0 ]["data" ][i ]["dataMap" ].update (fitstat_data )
1303
+ fitstat_data = None
1304
+ if roc_dict [j ]["dataMap" ]["_KS_" ] == 1 :
1305
+ fitstat_data = dict ()
1306
+ missing_stats = "_KS_" , "_KS2_" , "_C_" , "_Gini_" , "_Gamma_" , "_Tau_"
1307
+ for stat in missing_stats :
1308
+ if stat in roc_dict [j ]["dataMap" ]:
1309
+ fitstat_data [stat ] = roc_dict [j ]["dataMap" ][stat ]
1310
+ if fitstat_data :
1311
+ json_dict [0 ]["data" ][i ]["dataMap" ].update (fitstat_data )
1314
1312
1315
1313
lift_df = pd .DataFrame (conn .CASTable ("Lift" , caslib = "Public" ).to_frame ())
1316
1314
lift_dict = cls .apply_dataframe_to_json (json_dict [2 ]["data" ], i , lift_df , 1 )
0 commit comments