afdee1c/symbolic-regression
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
Silviu Marian Udrescu 2020-06-20 19:30:25 -04:00 • committed by GitHub
parent 7a0d782d51
commit a96e7f321a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
10 changed files with 119 additions and 133 deletions
Download patch file Download diff file Expand all files Collapse all files

6
Code/S_snap.py
View file

@ -60,14 +60,14 @@ def integerSnap(p, top=1):
p = np.array(p)
metric = np.abs(p - np.round(p.astype(np.double)))
chosen = np.argsort(metric)[:top]
return list(zip(chosen, np.round(p.astype(np.double))[chosen]))
return dict(list(zip(chosen, np.round(p.astype(np.double))[chosen])))
def zeroSnap(p, top=1):
p = np.array(p)
metric = np.abs(p)
chosen = np.argsort(metric)[:top]
return list(zip(chosen, np.zeros(len(chosen))))
return dict(list(zip(chosen, np.zeros(len(chosen)))))
def rationalSnap(p, top=1):
@ -75,4 +75,4 @@ def rationalSnap(p, top=1):
p = np.array(p)
snaps = np.array(list(bestApproximation(x,100) for x in p))
chosen = np.argsort(snaps[:, 3])[:top]
return list(zip(chosen, snaps[chosen, 0:3]))
return dict(list(zip(chosen, snaps[chosen, 0:3])))