change model path

TsumaR · Oct 7, 2024 · f484706 · f484706
1 parent d4701dd
commit f484706
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/vclean/__init__.py b/vclean/__init__.py
@@ -2,4 +2,4 @@
 vClean: Assessing the contamination of viral genomes
 """
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
diff --git a/vclean/modules/run_for_cli.py b/vclean/modules/run_for_cli.py
@@ -285,12 +285,12 @@ def main(args):
 
         script_dir = os.path.dirname(os.path.abspath(__file__))
         model_dir = os.path.dirname(os.path.dirname(script_dir))
-        model_path = os.path.join(model_dir, 'models', 'best_lgb_model_v1.3.2.pickle')
+        model_path = os.path.join(model_dir, 'vclean', 'models', 'best_lgb_model_v1.3.2.pickle')
 
         y_pred_proba_list = []
         for i in range(5):
             model_name = f'best_lgb_model_fold{i}.txt'
-            model_path = os.path.join(model_dir, 'models', model_name)
+            model_path = os.path.join(model_dir, 'vclean', 'models', model_name)
             loaded_model = lgb.Booster(model_file=model_path)
             y_pred_proba = loaded_model.predict(result_for_input)
             y_pred_proba_list.append(y_pred_proba)