Skip to content

Commit a183984

Browse files
committed
Prep for new handling of classification model types in Viya 4
1 parent ede6691 commit a183984

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/sasctl/pzmm/write_json_files.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -173,10 +173,13 @@ def writeModelPropertiesJSON(
173173
"WARNING: The provided model description was truncated to 1024 characters."
174174
)
175175

176-
if numTargetCategories > 2:
176+
if numTargetCategories > 2 and not targetEvent:
177177
targetLevel = "NOMINAL"
178+
elif numTargetCategories > 2 and targetEvent:
179+
targetLevel = "ORDINAL"
178180
else:
179181
targetLevel = "BINARY"
182+
targetEvent = 1
180183

181184
if eventProbVar is None:
182185
try:

0 commit comments

Comments
 (0)