ymlin105 commited on
Commit
205fba1
·
1 Parent(s): 92ff78b

fix: add model params logging for debugging

Browse files
Files changed (1) hide show
  1. src/app.py +12 -0
src/app.py CHANGED
@@ -58,6 +58,18 @@ def startup_event():
58
  pipeline.model = pipeline.model.set_params(device="cpu")
59
  # Force using hist tree method
60
  pipeline.model = pipeline.model.set_params(tree_method="hist")
 
 
 
 
 
 
 
 
 
 
 
 
61
  logger.info("Model parameters fixed for compatibility")
62
  except Exception as e:
63
  logger.warning(f"Could not fix model params: {e}")
 
58
  pipeline.model = pipeline.model.set_params(device="cpu")
59
  # Force using hist tree method
60
  pipeline.model = pipeline.model.set_params(tree_method="hist")
61
+
62
+ # Log model params
63
+ logger.info(
64
+ f"Model params: n_estimators={getattr(pipeline.model, 'n_estimators', 'N/A')}"
65
+ )
66
+ logger.info(
67
+ f"Model params: max_depth={getattr(pipeline.model, 'max_depth', 'N/A')}"
68
+ )
69
+ logger.info(
70
+ f"Model params: learning_rate={getattr(pipeline.model, 'learning_rate', 'N/A')}"
71
+ )
72
+
73
  logger.info("Model parameters fixed for compatibility")
74
  except Exception as e:
75
  logger.warning(f"Could not fix model params: {e}")