We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 15016a5 commit 718bbb5Copy full SHA for 718bbb5
autointent/modules/embedding/_logreg.py
@@ -77,8 +77,8 @@ def __init__(
77
def from_context(
78
cls,
79
context: Context,
80
- cv: int,
81
embedder_name: str,
+ cv: int = 3,
82
) -> "LogregAimedEmbedding":
83
"""
84
Create a LogregAimedEmbedding instance using a Context object.
@@ -89,8 +89,8 @@ def from_context(
89
:return: Initialized LogregAimedEmbedding instance.
90
91
return cls(
92
- cv=cv,
93
embedder_name=embedder_name,
+ cv=cv,
94
embedder_device=context.get_device(),
95
embedder_batch_size=context.get_batch_size(),
96
embedder_max_length=context.get_max_length(),
0 commit comments