We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 45e3a8a commit f74cee4Copy full SHA for f74cee4
1 file changed
mambular/models/sklearn_base_classifier.py
@@ -146,7 +146,7 @@ def build_model(
146
The validation target values. Required if `X_val` is provided.
147
random_state : int, default=101
148
Controls the shuffling applied to the data before applying the split.
149
- batch_size : int, default=64
+ batch_size : int, default=128
150
Number of samples per gradient update.
151
shuffle : bool, default=True
152
Whether to shuffle the training data before each epoch.
0 commit comments