From 3cbef17b0c40cc25082447fe0324524b2d7a2840 Mon Sep 17 00:00:00 2001 From: nuluh Date: Mon, 28 Jul 2025 05:20:10 +0700 Subject: [PATCH] feat(model_selection): add timing for model training and validation processes --- code/src/ml/model_selection.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/code/src/ml/model_selection.py b/code/src/ml/model_selection.py index 5243536..2c10474 100644 --- a/code/src/ml/model_selection.py +++ b/code/src/ml/model_selection.py @@ -117,11 +117,20 @@ def train_and_evaluate_model( result = {"model": model_name, "sensor": sensor_label, "success": False} try: + import time + start_time = time.time() + # Train the model model.fit(x_train, y_train) + result["elapsed_time_training"] = time.time() - start_time try: + # Predict on the test set (validation) + start_time = time.time() + y_pred = model.predict(x_test) + + result["elapsed_time_validation"] = time.time() - start_time result["y_pred"] = y_pred # Convert to numpy array except Exception as e: result["error"] = f"Prediction error: {str(e)}"