From db2cc6862f4cdd243d4fd0431b2287448a13414d Mon Sep 17 00:00:00 2001 From: sosokker Date: Sat, 11 May 2024 02:04:56 +0700 Subject: [PATCH] Change prediction input feature --- StreamServer/src/analytic/health/indoor_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/StreamServer/src/analytic/health/indoor_model.py b/StreamServer/src/analytic/health/indoor_model.py index 674e45b..87ad441 100644 --- a/StreamServer/src/analytic/health/indoor_model.py +++ b/StreamServer/src/analytic/health/indoor_model.py @@ -2,7 +2,7 @@ import joblib class XgboostIndoorModel: - MODEL_PATH = 'xgboost_model.pkl' + MODEL_PATH = 'analytic/health/xgboost_model.pkl' def __init__(self): self.__model = joblib.load(self.MODEL_PATH) @@ -13,7 +13,7 @@ class XgboostIndoorModel: Input order : ['outdoor_temp', 'outdoor_feels_like', 'outdoor_pressure', 'outdoor_humidity', 'outdoor_pm25', 'outdoor_pm10'] """ - if len(X) != 6: + if len(X) != 4: raise ValueError(f"Expected 6 features, got {len(X)}") return self.__model.predict([X])[0] \ No newline at end of file