fix: fixed _classifier

Author: Darinochka

autointent/modules/embedding/_logreg.py

@@ -113,14 +113,14 @@ def fit(self, utterances: list[str], labels: list[LabelType]) -> None:
         """
         self._multilabel = isinstance(labels[0], list)
 
-        self.embedder = Embedder(
+        self._embedder = Embedder(
             device=self.embedder_device,
             model_name_or_path=self.embedder_name,
             batch_size=self.embedder_batch_size,
             max_length=self.embedder_max_length,
             use_cache=self.embedder_use_cache,
         )
-        embeddings = self.embedder.embed(utterances)
+        embeddings = self._embedder.embed(utterances)
 
         if self._multilabel:
             self._label_encoder = None
@@ -155,7 +155,7 @@ def score(
             message = f"Invalid split '{split}' provided. Expected one of 'validation', or 'test'."
             raise ValueError(message)
 
-        embeddings = self.embedder.embed(utterances)
+        embeddings = self._embedder.embed(utterances)
         probas = self._classifier.predict_proba(embeddings)
         if self._multilabel:
             probas = np.stack(probas, axis=1)[..., 1]

tests/modules/retrieval/test_logreg.py

@@ -16,8 +16,8 @@ def test_fit_trains_model():
     labels = [0, 1, 0, 1, 1, 0, 0, 0, 0, 1, 1]
     module.fit(utterances, labels)
 
-    assert module.classifier.coef_ is not None
-    assert len(module.classifier.coef_) > 0
+    assert module._classifier.coef_ is not None
+    assert len(module._classifier.coef_) > 0
     assert module.label_encoder.classes_.tolist() == [0, 1]