Skip to content

Commit

Permalink
Remove print statements
Browse files Browse the repository at this point in the history
  • Loading branch information
vinayvinkumar authored May 9, 2023
1 parent 8fd6c44 commit 4550438
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions nyoka/xgboost/xgboost_to_pmml.py
Original file line number Diff line number Diff line change
Expand Up @@ -402,8 +402,6 @@ def create_node_new(root, derived_col_names, mining_fields, isroot = False):
if childs:
l = create_node_new(childs[0], derived_col_names, mining_fields)
r = create_node_new(childs[1], derived_col_names, mining_fields)
# print(root["split_condition"])
# print(stack)

temp = pml.Node(score=stack.pop())
temp.SimplePredicate = pml.SimplePredicate(field=replace_name_with_derivedColumnNames(root['split'], derived_col_names),
Expand All @@ -428,8 +426,6 @@ def create_node_new(root, derived_col_names, mining_fields, isroot = False):
l.append(temp)
return l

# print(stack)

return [temp]

else:
Expand Down

0 comments on commit 4550438

Please sign in to comment.