pyeggp 1.0.2__cp310-cp310-manylinux_2_28_aarch64.whl → 1.0.3__cp310-cp310-manylinux_2_28_aarch64.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
@@ -152,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.2.dist-info → pyeggp-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyeggp
-Version: 1.0.2
+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.2.dist-info → pyeggp-1.0.3.dist-info}/RECORD

@@ -1,6 +1,6 @@
-pyeggp/__init__.py,sha256=ag29g10K_goyZDXVPc-QeT5M332KN1mddYZ-FaEGlgQ,5775
+pyeggp/__init__.py,sha256=pD4mYwe9-JQ7vmD5zGdC3EtGfyj7o1xmetYBHdTOB5Q,6572
 pyeggp/__main__.py,sha256=dAMEebAw-mVcR5PUffuRsUDPFm6Vf4qvaam9TPSUs0I,155
-pyeggp/_binding.cpython-310-aarch64-linux-gnu.so,sha256=k15K_l1rO5R97zst-4cazmG1aSyMFVeBI6LHvH9-qG8,1016193
+pyeggp/_binding.cpython-310-aarch64-linux-gnu.so,sha256=qftUCZQJ1dKkgRYVk28bxpfUfYcaXY1MYEuDjOh-Joc,1016193
 pyeggp/_binding.py,sha256=nTO9o8SQC3SXK4zF3df5aBOQ4ktn2a-3P8gQTnMrNIQ,591
 pyeggp/_binding.pyi,sha256=TsI8Gar1wpDs3c8V26i3w_u1JjznlE5dReV59bbVUh0,460
 pyeggp/binding.i,sha256=AdPEymjqByW6OhpJS08o2oTc7Go3-274JqRhbXoyDjg,1908
@@ -119,9 +119,9 @@ pyeggp./libHSzlib-0-48ee1f9e.7.1.0-5585aebc199b08d8e21b5678c0fd28335affe6252f546
 pyeggp./libffi-c0132414.so.8,sha256=oTvjWevsiUzLHfgJ-ZZLqC1CD1h0cC6biH2mE8iWFuE,132737
 pyeggp./libgmp-6d1678ee.so.10.3.2,sha256=K-WFYhW6C3Z3ecUJJps22sV11vBCXEVqIP5tzap4xyg,605705
 pyeggp./libnlopt-38a103cb.so.0,sha256=SoGgo4hNv89sqJRH5RWvp21cEoJFfQCkLSkVTBynEp8,738297
-pyeggp-1.0.2.dist-info/METADATA,sha256=ezRmsnPEc0Uc8Coi_5xX08en30lyJdV0PKk8gT2njvc,42220
-pyeggp-1.0.2.dist-info/WHEEL,sha256=eDp1bqKv1fOPqy6hq3cz1RoeS_CSkkWg6r1puEBnKq4,114
-pyeggp-1.0.2.dist-info/entry_points.txt,sha256=NJouaQ2UoCLBu7Toqer0FBvNSDYMawR_Wx9STMdopyY,48
-pyeggp-1.0.2.dist-info/top_level.txt,sha256=iLrWfWZHh4NruYDgVyn2ntEB-uJ8fIs95nzNVLIhKE4,7
-pyeggp-1.0.2.dist-info/RECORD,,
-pyeggp-1.0.2.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+pyeggp-1.0.3.dist-info/METADATA,sha256=iubFYUSUPnA_cO2U9VqWB47FKamC7_WdTStNT67bl4o,42220
+pyeggp-1.0.3.dist-info/WHEEL,sha256=eDp1bqKv1fOPqy6hq3cz1RoeS_CSkkWg6r1puEBnKq4,114
+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/RECORD,,
+pyeggp-1.0.3.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149