pyeggp 1.0.1__cp39-cp39-macosx_14_0_x86_64.whl → 1.0.3__cp39-cp39-macosx_14_0_x86_64.whl

pyeggp/__init__.py

@@ -2,6 +2,7 @@ import atexit
 from contextlib import contextmanager
 from threading import Lock
 from typing import Iterator, List
+import string
 from io import StringIO
 import tempfile
 import csv
@@ -58,6 +59,11 @@ def pyeggp_run(dataset: str, gen: int, nPop: int, maxSize: int, nTournament: int
     with hs_rts_init():
         return unsafe_hs_pyeggp_run(dataset, gen, nPop, maxSize, nTournament, pc, pm, nonterminals, loss, optIter, optRepeat, nParams, split, simplify, dumpTo, loadFrom)
 
+def make_function(expression):
+    def func(x, t):
+        return eval(expression)
+    return func
+
 class PyEGGP(BaseEstimator, RegressorMixin):
     def __init__(self, gen = 100, nPop = 100, maxSize = 15, nTournament = 3, pc = 0.9, pm = 0.3, nonterminals = "add,sub,mul,div", loss = "MSE", optIter = 50, optRepeat = 2, nParams = -1, split = 1, simplify = False, dumpTo = "", loadFrom = ""):
         self.gen = gen
@@ -137,6 +143,7 @@ class PyEGGP(BaseEstimator, RegressorMixin):
         best = self.results[self.results.id==ix].iloc[view]
         t = np.array(list(map(float, best.theta.split(";"))))
         return eval(best.Numpy)
+
     def evaluate_model_view(self, x, ix, view):
         if x.ndim == 1:
             x = x.reshape(-1,1)
@@ -151,3 +158,19 @@ class PyEGGP(BaseEstimator, RegressorMixin):
     def score(self, X, y):
         ypred = self.evaluate_best_model(X)
         return r2_score(y, ypred)
+    def get_model(self, idx):
+        alphabet = list(string.ascii_uppercase)
+        row = self.results[self.results['id']==idx].iloc[0]
+        visual_expression = row['Numpy']
+        model = make_function(visual_expression)
+        n_params_used = len(row['theta'].split(sep=';'))
+    
+        # Works for solutions with less than 26 parameters
+        for i in range(n_params_used):
+            visual_expression = visual_expression.replace(f't[{i}]', alphabet[i])
+    
+        # Works for data with less than 50 dimensions
+        for i in range(50):
+            visual_expression = visual_expression.replace(f'x[:, {i}]', f'X{i}')
+    
+        return model, visual_expression

{pyeggp-1.0.1.dist-info → pyeggp-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyeggp
-Version: 1.0.1
+Version: 1.0.3
 Summary: Python Wheels for eggp algorithm.
 Author-email: Fabricio Olivetti <folivetti@users.noreply.github.com>
 License:                     GNU GENERAL PUBLIC LICENSE

{pyeggp-1.0.1.dist-info → pyeggp-1.0.3.dist-info}/RECORD

@@ -1,14 +1,14 @@
-pyeggp-1.0.1.dist-info/RECORD,,
-pyeggp-1.0.1.dist-info/WHEEL,sha256=SY5k3lEzlF5bfpGBmmxcfZtVPJk-byPddM4TNHg-8Nw,108
-pyeggp-1.0.1.dist-info/entry_points.txt,sha256=NJouaQ2UoCLBu7Toqer0FBvNSDYMawR_Wx9STMdopyY,48
-pyeggp-1.0.1.dist-info/top_level.txt,sha256=iLrWfWZHh4NruYDgVyn2ntEB-uJ8fIs95nzNVLIhKE4,7
-pyeggp-1.0.1.dist-info/METADATA,sha256=vbigkLKhJ_sSORsiKsZkFnU-Mjs8jOSN4Rj7gypQbtU,42220
-pyeggp-1.0.1.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-pyeggp/__init__.py,sha256=rzgGMRTJwRU0BDhfUCcE1G0OvFwu3o0qcjTfTRV7q54,5774
+pyeggp/__init__.py,sha256=pD4mYwe9-JQ7vmD5zGdC3EtGfyj7o1xmetYBHdTOB5Q,6572
 pyeggp/_binding.pyi,sha256=TsI8Gar1wpDs3c8V26i3w_u1JjznlE5dReV59bbVUh0,460
-pyeggp/_binding.cpython-39-darwin.so,sha256=shzLZ_G_gX4OHAJsApkLwJhDq0uyHZUIQOvCsQ3_nUY,55147408
+pyeggp/_binding.cpython-39-darwin.so,sha256=8-qCugl1wHf_hfJuzhHb6egOu0OpgoDxSDhmug1OxW0,55148144
 pyeggp/binding.i,sha256=AdPEymjqByW6OhpJS08o2oTc7Go3-274JqRhbXoyDjg,1908
 pyeggp/_binding.py,sha256=XImcPYlOIstr_dT9qMD68Ay7vPUB4mmduZ818hEKk3Q,579
 pyeggp/typing.py,sha256=qr1vWstl5HiaUT5HGsQFHO4pRq4EhqEc4zG2mn9QHgc,517
 pyeggp/__main__.py,sha256=dAMEebAw-mVcR5PUffuRsUDPFm6Vf4qvaam9TPSUs0I,155
 pyeggp/.dylibs/libnlopt.1.0.0.dylib,sha256=KLCzh-73JgicfAN8r0WwrJmSXeKsu498GZ1uTQiUu6g,480064
+pyeggp-1.0.3.dist-info/RECORD,,
+pyeggp-1.0.3.dist-info/WHEEL,sha256=SY5k3lEzlF5bfpGBmmxcfZtVPJk-byPddM4TNHg-8Nw,108
+pyeggp-1.0.3.dist-info/entry_points.txt,sha256=NJouaQ2UoCLBu7Toqer0FBvNSDYMawR_Wx9STMdopyY,48
+pyeggp-1.0.3.dist-info/top_level.txt,sha256=iLrWfWZHh4NruYDgVyn2ntEB-uJ8fIs95nzNVLIhKE4,7
+pyeggp-1.0.3.dist-info/METADATA,sha256=iubFYUSUPnA_cO2U9VqWB47FKamC7_WdTStNT67bl4o,42220
+pyeggp-1.0.3.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149