From 7f60cf8624606db87312df15af220e2cbb8611a2 Mon Sep 17 00:00:00 2001 From: chenyangkang Date: Wed, 29 May 2024 14:34:32 +0800 Subject: [PATCH] njobs of fit function defaults to njob attribute of object --- stemflow/model/AdaSTEM.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/stemflow/model/AdaSTEM.py b/stemflow/model/AdaSTEM.py index 9baa4d7..98d1eca 100644 --- a/stemflow/model/AdaSTEM.py +++ b/stemflow/model/AdaSTEM.py @@ -568,7 +568,7 @@ def fit( y_train: Union[pd.core.frame.DataFrame, np.ndarray], verbosity: Union[None, int] = None, ax=None, - njobs: int = 1, + njobs: Union[None, int] = None, ): """Fitting method @@ -576,6 +576,9 @@ def fit( X_train: Training variables y_train: Training target ax: matplotlib Axes to add to + verbosty: whether to show progress bar. 0 for no and 1 for yes. + ax: matplotlib ax for adding grid plot on that. + njobs: multiprocessing thread count. Default the njob of model object. Raises: TypeError: X_train is not a type of pd.core.frame.DataFrame