15 lines
370 B
Python
15 lines
370 B
Python
from sklearn.linear_model import LogisticRegression
|
|
|
|
|
|
class PixelClassifier:
|
|
def fit(self, X, y):
|
|
self._classifier = LogisticRegression(max_iter=1000)
|
|
self._classifier.fit(X, y)
|
|
return self
|
|
|
|
def predict(self, X):
|
|
return self._classifier.predict(X)
|
|
|
|
def predict_proba(self, X):
|
|
return self._classifier.predict_proba(X)
|