From 95e7baeb889c8d6ce9dcb6d638d8aaf2ec373c64 Mon Sep 17 00:00:00 2001 From: Soham Date: Sun, 15 Feb 2026 23:23:16 +0530 Subject: [PATCH] Fix inconsistent model training approach in gaze tracker --- app/services/gaze_tracker.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/app/services/gaze_tracker.py b/app/services/gaze_tracker.py index 3354551..f3bd872 100644 --- a/app/services/gaze_tracker.py +++ b/app/services/gaze_tracker.py @@ -87,8 +87,7 @@ def trian_and_predict(model_name, X_train, y_train, X_test, y_test, label): """ if ( model_name == "Linear Regression" - or model_name == "Elastic Net" - or model_name == "Support Vector Regressor" + or model_name == "Random Forest Regressor" ): model = models[model_name] model.fit(X_train, y_train)