Skip to content

Commit 36d704c

Browse files
committed
fix: added cv to the docs example
1 parent 34fb1ca commit 36d704c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

autointent/modules/embedding/_retrieval.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ class LogRegEmbedding(EmbeddingModule):
4444
retrieval = LogRegEmbedding(
4545
k=3,
4646
embedder_name="sergeyzh/rubert-tiny-turbo",
47+
cv=2
4748
)
4849
retrieval.fit(utterances, labels)
4950
"""

0 commit comments

Comments
 (0)