Browse Source

[MNT] resolve comments in abl_model.py

pull/1/head
Gao Enhao 2 years ago
parent
commit
1fc02dc539
1 changed files with 10 additions and 1 deletions
  1. +10
    -1
      abl/learning/abl_model.py

+ 10
- 1
abl/learning/abl_model.py View File

@@ -12,7 +12,10 @@ class ABLModel:
Parameters
----------
base_model : Machine Learning Model
The base model to use for training and prediction.
The base machine learning model used for training and prediction. This model should
implement the 'fit' and 'predict' methods. It's recommended, but not required,for the
model to also implement the 'predict_proba' method for generating probabilistic
predictions.
"""

def __init__(self, base_model: Any) -> None:
@@ -115,11 +118,17 @@ class ABLModel:
def save(self, *args, **kwargs) -> None:
"""
Save the model to a file.

This method delegates to the 'save' method of self.base_model. The arguments passed to
this method should match those expected by the 'save' method of self.base_model.
"""
self._model_operation("save", *args, **kwargs)

def load(self, *args, **kwargs) -> None:
"""
Load the model from a file.

This method delegates to the 'load' method of self.base_model. The arguments passed to
this method should match those expected by the 'load' method of self.base_model.
"""
self._model_operation("load", *args, **kwargs)

Loading…
Cancel
Save