Revisions

2026-06-08 15:15:52 -04:00
parent 1c99ca8bd3
commit 49c4e43f45
12 changed files with 209 additions and 34 deletions
--- a/models/pixels.py
+++ b/models/pixels.py
@@ -9,3 +9,6 @@ class PixelClassifier:

    def predict(self, X):
        return self._classifier.predict(X)
+
+    def predict_proba(self, X):
+        return self._classifier.predict_proba(X)