Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

njobs of fit function defaults to njob attribute of object #52

Merged
merged 1 commit into from
May 29, 2024
Merged

njobs of fit function defaults to njob attribute of object #52

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion stemflow/model/AdaSTEM.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -568,14 +568,17 @@ 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

Args:
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
Expand Down
Loading