Skip to content

Commit e09d42c

Browse files
committed
seed -> random_seed
1 parent 77d1886 commit e09d42c

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

‎model2vec/train/classifier.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ def fit( # noqa: C901 # Complexity is bad.
158158
X_val: list[str] | None = None,
159159
y_val: LabelType | None = None,
160160
class_weight: torch.Tensor | None = None,
161-
seed: int = _DEFAULT_RANDOM_SEED,
161+
random_seed: int = _DEFAULT_RANDOM_SEED,
162162
) -> StaticModelForClassification:
163163
"""
164164
Fit a model.
@@ -188,11 +188,11 @@ def fit( # noqa: C901 # Complexity is bad.
188188
:param y_val: The labels to be used for validation.
189189
:param class_weight: The weight of the classes. If None, all classes are weighted equally. Must
190190
have the same length as the number of classes.
191-
:param seed: The random seed to use. Defaults to 42.
191+
:param random_seed: The random seed to use. Defaults to 42.
192192
:return: The fitted model.
193193
:raises ValueError: If either X_val or y_val are provided, but not both.
194194
"""
195-
pl.seed_everything(seed)
195+
pl.seed_everything(random_seed)
196196
logger.info("Re-initializing model.")
197197

198198
# Determine whether the task is multilabel based on the type of y.

0 commit comments

Comments
 (0)