Skip to content

Commit 3036e31

Browse files
committed
remove k completely
1 parent cf5fdef commit 3036e31

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed
Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,13 @@
22

33

44
def test_get_assets_returns_correct_artifact_for_logreg():
5-
module = LogregAimedEmbedding(k=5, embedder_name="sergeyzh/rubert-tiny-turbo")
5+
module = LogregAimedEmbedding(embedder_name="sergeyzh/rubert-tiny-turbo")
66
artifact = module.get_assets()
77
assert artifact.embedder_name == "sergeyzh/rubert-tiny-turbo"
88

99

1010
def test_fit_trains_model():
11-
module = LogregAimedEmbedding(k=5, embedder_name="sergeyzh/rubert-tiny-turbo")
11+
module = LogregAimedEmbedding(embedder_name="sergeyzh/rubert-tiny-turbo")
1212

1313
utterances = ["hello", "goodbye", "hi", "bye", "bye", "hello", "welcome", "hi123", "hiii", "bye-bye", "bye!"]
1414
labels = [0, 1, 0, 1, 1, 0, 0, 0, 0, 1, 1]
@@ -20,7 +20,7 @@ def test_fit_trains_model():
2020

2121

2222
def test_predict_evaluates_model():
23-
module = LogregAimedEmbedding(k=5, embedder_name="sergeyzh/rubert-tiny-turbo")
23+
module = LogregAimedEmbedding(embedder_name="sergeyzh/rubert-tiny-turbo")
2424

2525
utterances = ["hello", "goodbye", "hi", "bye", "bye", "hello", "welcome", "hi123", "hiii", "bye-bye", "bye!"]
2626
labels = [0, 1, 0, 1, 1, 0, 0, 0, 0, 1, 1]

0 commit comments

Comments
 (0)