xgboost

2026-04-25 03:04:52 +09:00
parent 8a8c0c9713
commit da4201fb20
11 changed files with 17829 additions and 105 deletions
--- a/predict/predict_xgboost.py
+++ b/predict/predict_xgboost.py
@@ -39,15 +39,17 @@ def safe_float(value):

 def predict(
    working_dir: str,
-    songno: str
+    songno: str,
+    feature: str = None
 ):
    # =====================================================
    # 경로
    # =====================================================

-    features_path = os.path.join(
-        working_dir,
-        FEATURES_FILENAME
+    features_path = (
+        os.path.join(working_dir, FEATURES_FILENAME)
+        if feature is None
+        else feature
    )

    model_path = os.path.join(
@@ -141,6 +143,11 @@ if __name__ == "__main__":
        "--workingDir",
        required=True
    )
+    
+    parser.add_argument(
+        "--feature",
+        required=False
+    )

    parser.add_argument(
        "--songno",