2024/03/09/11:45

analysis/model_train/tree_model.py

@@ -99,12 +99,21 @@ class DecisionTreeClassifierParams:
 
 # 决策树分类
 def decision_tree_classifier(container, params):
+    import logging
+    logging.basicConfig(level=logging.NOTSET)
+    logging.info(str(params), logging.getLevelName(logging.INFO))
+    print(str(params))
+
+
     x_train, y_train, x_test, y_test, hyper_params_optimize = get_values_from_container_class(container)
     info = {}
 
     params = transform_params_list(DecisionTreeClassifierParams, params)
     params['random_state'] = [StaticValue.RANDOM_STATE]
 
+    logging.info(str(params), logging.getLevelName(logging.INFO))
+    print(str(params))
+
     random_forest_regression_model = DecisionTreeClassifier(random_state=StaticValue.RANDOM_STATE)
 
     if hyper_params_optimize == "grid_search":