From 67944a8d40b44518bf501f92e099c13da8a8e149 Mon Sep 17 00:00:00 2001 From: vasiliev Date: Wed, 1 Jan 2025 20:05:54 +0300 Subject: [PATCH] Cleared code --- demonstration/Dissertation/memory.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/demonstration/Dissertation/memory.py b/demonstration/Dissertation/memory.py index 039e9ea..0f87bc2 100644 --- a/demonstration/Dissertation/memory.py +++ b/demonstration/Dissertation/memory.py @@ -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() \ No newline at end of file + train_model()