Initial commit
This commit is contained in:
11
models/pixels.py
Normal file
11
models/pixels.py
Normal file
@@ -0,0 +1,11 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user