tlemagueresse commited on
Commit
39fb66e
·
1 Parent(s): 02c69cc

Rename txt model file

Browse files
fast_model.py CHANGED
@@ -522,7 +522,7 @@ def save_pipeline(
522
  pipeline_file_name : str, optional
523
  The filename for saving the pipeline (default is "pipeline.pkl").
524
  """
525
- lgbm_file_name = lgbm_file_name or "model_fast_model.txt"
526
  pipeline_file_name = pipeline_file_name or "pipeline.pkl"
527
 
528
  lightgbm_path = Path(path) / lgbm_file_name
 
522
  pipeline_file_name : str, optional
523
  The filename for saving the pipeline (default is "pipeline.pkl").
524
  """
525
+ lgbm_file_name = lgbm_file_name or "model_lightgbm.txt"
526
  pipeline_file_name = pipeline_file_name or "pipeline.pkl"
527
 
528
  lightgbm_path = Path(path) / lgbm_file_name
model_fast_model.txt → model_lightgbm.txt RENAMED
File without changes