nuluh/thesis
1
0
Fork 0
Code Issues 47 Pull Requests Actions Packages Projects Releases 1 Wiki Activity

feat(ml): add XGBoost model to inference options and update commented inference calls

Browse Source
This commit is contained in:
nuluh
2025-06-27 10:35:27 +07:00
parent 465ed121f9
commit c2df42cc2b
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
code/src/ml/inference.py
View File

@@ -1,6 +1,9 @@
from src.ml.model_selection import inference_model from src.ml.model_selection import inference_model
model = {"SVM": "D:/thesis/models/sensor1/SVM.joblib", 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) # 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)