Spaces:
Sleeping
Sleeping
fix: add model params logging for debugging
Browse files- 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}")
|