Skip to content

Commit 48d2ab5

Browse files
committed
Added changes to fitstat for classification models
1 parent 4b7ebca commit 48d2ab5

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

src/sasctl/pzmm/write_json_files.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1299,6 +1299,16 @@ def calculate_model_statistics(
12991299
roc_dict = cls.apply_dataframe_to_json(json_dict[1]["data"], i, roc_df)
13001300
for j in range(len(roc_dict)):
13011301
json_dict[1]["data"][j].update(roc_dict[j])
1302+
if(roc_dict[j]["dataMap"]["_KS_"] == 1):
1303+
fitstat_data = {
1304+
"_KS_": roc_dict[j]["dataMap"]["_KS_"],
1305+
"_KS2_": roc_dict[j]["dataMap"]["_KS2_"],
1306+
"_C_": roc_dict[j]["dataMap"]["_C_"],
1307+
"_Gini_": roc_dict[j]["dataMap"]["_Gini_"],
1308+
"_Gamma_": roc_dict[j]["dataMap"]["_Gamma_"],
1309+
"_Tau_": roc_dict[j]["dataMap"]["_Tau_"]
1310+
}
1311+
json_dict[0]["data"][i]["dataMap"].update(fitstat_data)
13021312

13031313
lift_df = pd.DataFrame(conn.CASTable("Lift", caslib="Public").to_frame())
13041314
lift_dict = cls.apply_dataframe_to_json(json_dict[2]["data"], i, lift_df, 1)

0 commit comments

Comments
 (0)