Merge remote-tracking branch 'origin/dev' into dev

climate_health/predictor/poisson.py

@@ -0,0 +1,19 @@
+from climate_health.predictor.protocol import Predictor
+import sklearn.linear_model as lm
+
+
+class Poisson():
+    def __init__(self, alpha=1, fit_intercept=True):
+        self.model = lm.PoissonRegressor(alpha=alpha, fit_intercept=fit_intercept)
+
+    def predict(self, data):
+        self.model.predict(data)
+
+    def evaluate(self, data):
+        self.model.score(data, data)
+
+    def train(self, data):
+        self.model.fit(data)
+
+# def evaluate(predictor: Predictor, data):
+#     return predictor.evaluate(data)

climate_health/predictor/predictor.py → climate_health/predictor/protocol.py

@@ -11,8 +11,5 @@ def predict(self, data):
     def evaluate(self, data):
         pass
 
-    def save(self, path):
-        pass
-
-    def load(self, path):
+    def train(self, data):
         pass

requirements_dev.txt

@@ -8,3 +8,5 @@ hypothesis
 coverage
 pytest-cov
 build
+scikit-learn
+numpy