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 f092999 commit 5a641a6Copy full SHA for 5a641a6
tests/modules/scoring/test_description.py
@@ -17,7 +17,7 @@ def test_description_scorer(dataset, expected_prediction, multilabel):
17
db_dir, dump_dir, logs_dir = setup_environment()
18
data_handler = DataHandler(dataset, force_multilabel=multilabel)
19
20
- scorer = DescriptionScorer(embedder_name="sergeyzh/rubert-tiny-turbo", db_dir=db_dir, temperature=0.3, device="cpu")
+ scorer = DescriptionScorer(embedder_name="sergeyzh/rubert-tiny-turbo", temperature=0.3, device="cpu")
21
22
scorer.fit(data_handler.utterances_train, data_handler.labels_train, data_handler.label_description)
23
assert scorer.description_vectors.shape[0] == len(data_handler.label_description)
0 commit comments