From d93bab5c0fa66e31c628f14b4ed84b30d5f54750 Mon Sep 17 00:00:00 2001 From: "Edwin (Ed) Onuonga" Date: Tue, 2 Apr 2024 10:09:45 +0100 Subject: [PATCH] fix: use log probs for `KNNClassifier.predict_log_proba` (#247) --- sequentia/models/knn/classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sequentia/models/knn/classifier.py b/sequentia/models/knn/classifier.py index 4d6a416..51af161 100644 --- a/sequentia/models/knn/classifier.py +++ b/sequentia/models/knn/classifier.py @@ -297,7 +297,7 @@ def predict_log_proba( ----- This method requires a trained classifier — see :func:`fit`. """ - return np.log(self.predict_scores(X, lengths=lengths)) + return np.log(self.predict_proba(X, lengths=lengths)) @_validation.requires_fit def predict_proba(