Skip to content

Commit 718bbb5

Browse files
authored
Update _logreg.py (#120)
1 parent 15016a5 commit 718bbb5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

autointent/modules/embedding/_logreg.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,8 @@ def __init__(
7777
def from_context(
7878
cls,
7979
context: Context,
80-
cv: int,
8180
embedder_name: str,
81+
cv: int = 3,
8282
) -> "LogregAimedEmbedding":
8383
"""
8484
Create a LogregAimedEmbedding instance using a Context object.
@@ -89,8 +89,8 @@ def from_context(
8989
:return: Initialized LogregAimedEmbedding instance.
9090
"""
9191
return cls(
92-
cv=cv,
9392
embedder_name=embedder_name,
93+
cv=cv,
9494
embedder_device=context.get_device(),
9595
embedder_batch_size=context.get_batch_size(),
9696
embedder_max_length=context.get_max_length(),

0 commit comments

Comments
 (0)