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 5d4a8c9 commit 67339a1Copy full SHA for 67339a1
autointent/modules/embedding/_logreg.py
@@ -168,7 +168,7 @@ def get_assets(self) -> RetrieverArtifact:
168
"""
169
return RetrieverArtifact(embedder_name=self.embedder_name)
170
171
- def predict(self, utterances: list[str]) -> NDArray[np.float64] | list[NDArray[np.float64]]:
+ def predict(self, utterances: list[str]) -> NDArray[np.float64]:
172
embeddings = self._embedder.embed(utterances)
173
probas = self._classifier.predict_proba(embeddings)
174
0 commit comments