initial commit, implemented (#111)

royerlab · Mar 21, 2022 · 8ec016a · 8ec016a
1 parent c19595f
commit 8ec016a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aydin/regression/lgbm.py b/aydin/regression/lgbm.py
@@ -205,7 +205,7 @@ def __init__(self, model, gpu_prediction, loss_history):
     def _save_internals(self, path: str):
         if self.model is not None:
             lgbm_model_file = join(path, 'lgbm_model.txt')
-            self.model.save(lgbm_model_file)
+            self.model.save_model(lgbm_model_file)
 
     def _load_internals(self, path: str):
         lgbm_model_file = join(path, 'lgbm_model.txt')