psyke 0.10.1.dev11__py3-none-any.whl → 0.10.2.dev10__py3-none-any.whl

psyke/genetic/gin/__init__.py

@@ -32,6 +32,7 @@ class GIn:
         self.toolbox = None
         self.stats = None
         self.hof = None
+        self.best = None
 
         self.setup(warm)
 
@@ -70,8 +71,11 @@ class GIn:
             return accuracy_score(true, pred)
         raise NameError('Supported metrics are R2, MAE, MSE, F1, ACC')
 
-    def evaluate(self, individual):
-        to_pred, true = self.valid or (self.X, self.y)
+    def predict(self, to_pred):
+        return self.__predict(to_pred=to_pred)[0]
+
+    def __predict(self, individual=None, to_pred=None):
+        individual = individual or self.best
         boundaries = np.cumsum([0] + list(self.slices))
         cuts = [sorted(individual[boundaries[i]:boundaries[i + 1]]) for i in range(len(self.slices))]
 
@@ -79,7 +83,7 @@ class GIn:
         regionsT = self.region(self.X, cuts)
 
         y_pred = np.empty(len(to_pred), dtype=f'U{self.y.str.len().max()}') if self.output == Target.CLASSIFICATION \
-            else np.zeros_like(self.y)
+            else np.zeros(len(to_pred))
         valid_regions = 0
 
         for r in range(np.prod([s + 1 for s in self.slices])):
@@ -92,10 +96,13 @@ class GIn:
             y_pred[mask] = self.__output_estimation(maskT, to_pred[mask])
             valid_regions += 1
 
+        return y_pred, valid_regions
+
+    def evaluate(self, individual=None):
+        y_pred, valid_regions = self.__predict(individual or self.best, self.X if self.valid is None else self.valid[0])
         if valid_regions < self.min_rules:
             return -9999,
-
-        return self.__score(true, y_pred),
+        return self.__score(self.y if self.valid is None else self.valid[1], y_pred),
 
     def setup(self, warm=False):
         if not warm:
@@ -131,5 +138,5 @@ class GIn:
         pop = self.toolbox.population(n=n_pop)
         result, log = algorithms.eaSimple(pop, self.toolbox, cxpb=cxpb, mutpb=mutpb, ngen=n_gen,
                                           stats=self.stats, halloffame=self.hof, verbose=False)
-        best = tools.selBest(pop, 1)[0]
-        return best, self.evaluate(best)[0], result, log
+        self.best = tools.selBest(pop, 1)[0]
+        return self.best, self.evaluate()[0], result, log

{psyke-0.10.1.dev11.dist-info → psyke-0.10.2.dev10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: psyke
-Version: 0.10.1.dev11
+Version: 0.10.2.dev10
 Summary: Python-based implementation of PSyKE, i.e. a Platform for Symbolic Knowledge Extraction
 Home-page: https://github.com/psykei/psyke-python
 Author: Matteo Magnini

{psyke-0.10.1.dev11.dist-info → psyke-0.10.2.dev10.dist-info}/RECORD

@@ -26,7 +26,7 @@ psyke/extraction/real/utils.py,sha256=4NNL15Eu7cmkG9b29GBP6CKgMTV1cmiJVS0k1MbWpI
 psyke/extraction/trepan/__init__.py,sha256=H8F_wpFLPcfyx2tgOOno8FwUomxfVxVl1vxlb0ClP1g,6931
 psyke/extraction/trepan/utils.py,sha256=iSUJ1ooNQT_VO1KfBZuIUeUsyUbGdQf_pSEE87vMeQg,2320
 psyke/genetic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-psyke/genetic/gin/__init__.py,sha256=9lwTfz6yrtxzg_BKE-bogEeHNpyTMINZOP4rUvItfQM,5405
+psyke/genetic/gin/__init__.py,sha256=svO8hwklJ96ZuMiB0qbwBH1B096HvymP5llVt0ntSow,5772
 psyke/schema/__init__.py,sha256=axv4ejZY0ItUwrC9IXb_yAhaQL5f1vwvXXmaIAHJmt0,26063
 psyke/tuning/__init__.py,sha256=yd_ForFmHeYbtRXltY1fOa-mPJvpE6ijzg50M_8Sdxw,3649
 psyke/tuning/crash/__init__.py,sha256=zIHEF75EFy_mRIieqzP04qKLG3GLsSc_mYZHpPfkzxU,2623
@@ -38,8 +38,8 @@ psyke/utils/logic.py,sha256=ioP25WMTYNYEzaRDNDe3kGNWqZ6DA_63t19d-ky_2kM,12227
 psyke/utils/metrics.py,sha256=Oo5BOonOSfo0qYsXWT5dmypZ7jiStByFC2MKEU0uMHg,2250
 psyke/utils/plot.py,sha256=dE8JJ6tQ0Ezosid-r2jqAisREjFe5LqExRzsVi5Ns-c,7785
 psyke/utils/sorted.py,sha256=C3CPW2JisND30BRk5c1sAAHs3Lb_wsRB2qZrYFuRnfM,678
-psyke-0.10.1.dev11.dist-info/licenses/LICENSE,sha256=G3mPaubObvkBXbsgTTeYGLk_pNEW8tc7HZr4u_wLEpU,11398
-psyke-0.10.1.dev11.dist-info/METADATA,sha256=6xsiKSkxqfQSecVRo5uyh3IXBWFMl-vE_msJ7mlLrSU,8396
-psyke-0.10.1.dev11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-psyke-0.10.1.dev11.dist-info/top_level.txt,sha256=q1HglxOqqoIRukFtyis_ZNHczZg4gANRUPWkD7HAUTU,6
-psyke-0.10.1.dev11.dist-info/RECORD,,
+psyke-0.10.2.dev10.dist-info/licenses/LICENSE,sha256=G3mPaubObvkBXbsgTTeYGLk_pNEW8tc7HZr4u_wLEpU,11398
+psyke-0.10.2.dev10.dist-info/METADATA,sha256=DOa6rC0lTAZje50Bq2h_za8XPa1etUCy2zI6qDynPxE,8396
+psyke-0.10.2.dev10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+psyke-0.10.2.dev10.dist-info/top_level.txt,sha256=q1HglxOqqoIRukFtyis_ZNHczZg4gANRUPWkD7HAUTU,6
+psyke-0.10.2.dev10.dist-info/RECORD,,