xgboost

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",