From c2df42cc2b928ea8af6d84b610fd53a00bf021ff Mon Sep 17 00:00:00 2001 From: nuluh Date: Fri, 27 Jun 2025 10:35:27 +0700 Subject: [PATCH] feat(ml): add XGBoost model to inference options and update commented inference calls --- code/src/ml/inference.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/code/src/ml/inference.py b/code/src/ml/inference.py index cdc6662..bf28fc7 100644 --- a/code/src/ml/inference.py +++ b/code/src/ml/inference.py @@ -1,6 +1,9 @@ from src.ml.model_selection import inference_model model = {"SVM": "D:/thesis/models/sensor1/SVM.joblib", - "SVM with PCA": "D:/thesis/models/sensor1/SVM with StandardScaler and PCA.joblib"} + "SVM with PCA": "D:/thesis/models/sensor1/SVM with StandardScaler and PCA.joblib", + "XGBoost": "D:/thesis/models/sensor1/XGBoost.joblib"} -inference_model(model["SVM"], "D:/thesis/data/dataset_A/zzzAD2.TXT", column_question=1) \ No newline at end of file +# inference_model(model["SVM"], "D:/thesis/data/dataset_A/zzzAD10.TXT", column_question=10) +# inference_model(model["SVM with PCA"], "D:/thesis/data/dataset_A/zzzAD10.TXT", column_question=10) +inference_model(model["XGBoost"], "D:/thesis/data/dataset_A/zzzAD30.TXT", column_question=30) \ No newline at end of file