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 f36359a commit a8eac23Copy full SHA for a8eac23
autointent/modules/scoring/_sklearn/sklearn_scorer.py
@@ -128,15 +128,7 @@ def fit(
128
"""
129
self._validate_task(labels)
130
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
+ embedder = Embedder(embedder_config=self.embedder_config)
140
features = embedder.embed(utterances, TaskTypeEnum.classification)
141
142
clf = MultiOutputClassifier(self._base_clf) if self._multilabel else self._base_clf
0 commit comments