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)