Skip to content

Commit f74cee4

Browse files
committed
docstring fixed for default batch size
1 parent 45e3a8a commit f74cee4

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

mambular/models/sklearn_base_classifier.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def build_model(
146146
The validation target values. Required if `X_val` is provided.
147147
random_state : int, default=101
148148
Controls the shuffling applied to the data before applying the split.
149-
batch_size : int, default=64
149+
batch_size : int, default=128
150150
Number of samples per gradient update.
151151
shuffle : bool, default=True
152152
Whether to shuffle the training data before each epoch.

0 commit comments

Comments
 (0)