Skip to content

Commit

Permalink
Cleared code
Browse files Browse the repository at this point in the history
  • Loading branch information
iuliivasilev committed Jan 1, 2025
1 parent 47401d2 commit 67944a8
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion demonstration/Dissertation/memory.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,9 +50,11 @@
'ens_metric_name': 'IBS_REMAIN', 'l_reg': 0.0, 'leaf_model': 'base_zero_after',
'max_features': 0.9, 'min_samples_leaf': 0.01, 'n_jobs': 1, 'n_jobs_loop': 1, 'signif': 1, 'woe': True}


def train_model():
tree = CRAID(**p)
tree.fit(X_tr, y_tr)


if __name__ == "__main__":
train_model()
train_model()

0 comments on commit 67944a8

Please sign in to comment.