Skip to content

Commit a8eac23

Browse files
authored
Fix/sklearn scorer embedder (#244)
* fix * formatter
1 parent f36359a commit a8eac23

File tree

1 file changed

+1
-9
lines changed

1 file changed

+1
-9
lines changed

autointent/modules/scoring/_sklearn/sklearn_scorer.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -128,15 +128,7 @@ def fit(
128128
"""
129129
self._validate_task(labels)
130130

131-
embedder = Embedder(
132-
EmbedderConfig(
133-
model_name=self.embedder_config.model_name,
134-
device=self.embedder_config.device,
135-
batch_size=self.embedder_config.batch_size,
136-
tokenizer_config=self.embedder_config.tokenizer_config,
137-
use_cache=self.embedder_config.use_cache,
138-
)
139-
)
131+
embedder = Embedder(embedder_config=self.embedder_config)
140132
features = embedder.embed(utterances, TaskTypeEnum.classification)
141133

142134
clf = MultiOutputClassifier(self._base_clf) if self._multilabel else self._base_clf

0 commit comments

Comments
 (0)