|
|
import sympy as sp |
|
|
import numpy as np |
|
|
import warnings |
|
|
from sympy.abc import x |
|
|
import sys |
|
|
import json |
|
|
from tqdm import tqdm |
|
|
|
|
|
from remend.tools.parser import parse_prefix_to_sympy |
|
|
|
|
|
warnings.simplefilter("ignore") |
|
|
|
|
|
def percent(a, n): |
|
|
return f"{a/n*100:0.1f}%" |
|
|
|
|
|
def do_eval_match(orig_expr, gen_expr): |
|
|
try: |
|
|
origl = sp.lambdify(x, orig_expr) |
|
|
genl = sp.lambdify(x, gen_expr) |
|
|
count = 0 |
|
|
|
|
|
for v in np.arange(0.2, 1, 0.01): |
|
|
o = origl(v) |
|
|
g = genl(v) |
|
|
if o != o or o == float('inf'): |
|
|
continue |
|
|
if g != g or g == float('inf'): |
|
|
continue |
|
|
|
|
|
|
|
|
|
|
|
if abs((o-g)/o) > 1e-5: |
|
|
return False |
|
|
count += 1 |
|
|
except: |
|
|
return False |
|
|
return count >= 5 |
|
|
|
|
|
if __name__ == "__main__": |
|
|
import argparse |
|
|
parser = argparse.ArgumentParser("Check generated expressions") |
|
|
parser.add_argument("-g", required=True, help="Generated expressions file") |
|
|
parser.add_argument("-i", required=True, help="Info file") |
|
|
parser.add_argument("-r", required=True, help="Results file") |
|
|
args = parser.parse_args() |
|
|
|
|
|
gens = [] |
|
|
with open(args.g, 'r') as genf, open(args.i) as infof: |
|
|
for line in tqdm(genf, desc="Reading file"): |
|
|
comps = line.strip().split("\t") |
|
|
if line[0] == 'H': |
|
|
num = int(comps[0][2:]) |
|
|
tokens = comps[2].split(" ") |
|
|
info = next(infof) |
|
|
info = json.loads(info.strip()) |
|
|
if info["eqn"] == "": |
|
|
continue |
|
|
gens.append((num, tokens, info)) |
|
|
|
|
|
parsed = [] |
|
|
matched = [] |
|
|
results = [] |
|
|
|
|
|
for n, toks, info in tqdm(gens, desc="Evaluating expressions"): |
|
|
res = {"id": n, "parsed": False, "matched": False, "orig": "", "gen": ""} |
|
|
if "<<unk>>" in toks: |
|
|
|
|
|
results.append(res) |
|
|
continue |
|
|
try: |
|
|
gen_expr = parse_prefix_to_sympy(toks) |
|
|
except Exception as e: |
|
|
|
|
|
results.append(res) |
|
|
continue |
|
|
|
|
|
res["parsed"] = True |
|
|
parsed.append(n) |
|
|
const = info["constants"] |
|
|
|
|
|
gen_expr = gen_expr.subs([(sp.Symbol("k"+c), const[c]) for c in const]) |
|
|
orig_expr = sp.parse_expr(info["eqn"], local_dict={"x0":x}) |
|
|
res["orig"] = str(orig_expr) |
|
|
res["gen"] = str(gen_expr) |
|
|
|
|
|
if not do_eval_match(orig_expr, gen_expr): |
|
|
results.append(res) |
|
|
continue |
|
|
res["matched"] = True |
|
|
matched.append(n) |
|
|
results.append(res) |
|
|
|
|
|
with open(args.r, "w") as resf: |
|
|
for res in results: |
|
|
resf.write("{id} {parsed} {matched} \"{orig}\" \"{gen}\"\n".format(**res)) |
|
|
resf.write("\n") |
|
|
N = len(gens) |
|
|
print("Total", N, file=resf) |
|
|
print("Parsed", len(parsed), percent(len(parsed), N), file=resf) |
|
|
print("Matched", len(matched), percent(len(matched), N), file=resf) |
|
|
|
