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-04-21 00:35:17 -04:00 • committed by GitHub
parent c77f86e80a
commit 1c7765c961
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
Code/RPN_to_pytorch.py
View file

@ -122,9 +122,12 @@ def RPN_to_pytorch(data_file, math_expr, lr = 1e-2, N_epochs = 500):
else: else:
eq = eq.subs(parm, trainable_parameters[ii]) eq = eq.subs(parm, trainable_parameters[ii])
complexity = complexity + get_number_DL(trainable_parameters[ii].detach().numpy()) complexity = complexity + get_number_DL(trainable_parameters[ii].detach().numpy())
n_variables = len(eq.free_symbols)
n_operations = len(count_ops(eq,visual=True).free_symbols)
if n_operations!=0 or n_variables!=0:
complexity = complexity + (n_variables+n_operations)*np.log2((n_variables+n_operations))
ii = ii+1 ii = ii+1
error = torch.mean((f(*input)-y)**2).data.numpy()*1 error = torch.mean((f(*input)-y)**2).data.numpy()*1
return error, complexity, eq return error, complexity, eq