File size: 2,677 Bytes
451da7d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
# augmentor.py
import random
import re
ALL_OPERANDS = ['+', '-', '*', '/', 'log', 'exp', 'cos', 'sqrt', 'asin', 'sin', '**', 'tan', 'abs']
def extract_operators(expr_str):
ops = set()
if 'exp' in expr_str: ops.add('exp')
if 'log' in expr_str: ops.add('log')
if 'cos' in expr_str: ops.add('cos')
if 'sin' in expr_str: ops.add('sin')
if '**' in expr_str: ops.add('**')
if 'sqrt' in expr_str: ops.add('sqrt')
if 'asin' in expr_str: ops.add('asin')
if 'tan' in expr_str: ops.add('tan')
if 'abs' in expr_str: ops.add('abs')
if '/' in expr_str: ops.add('/')
for op in ['+', '-', '*']:
if op in expr_str: ops.add(op)
return list(ops)
def infer_max_var(expr_str):
matches = re.findall(r'x_(\d+)', expr_str)
return max([int(m) for m in matches]) if matches else 1
def generate_expression_instructions(expr_str):
max_var = infer_max_var(expr_str)
variables = [f"x_{i}" for i in range(1, max_var + random.randint(1, (max_var) + 1))]
used_ops = extract_operators(expr_str)
extra_ops = list(set(ALL_OPERANDS) - set(used_ops))
added_ops = random.sample(extra_ops, random.randint(1, len(extra_ops))) if extra_ops else []
all_ops = sorted(set(used_ops + added_ops))
constants = ['C']
wrapped_expr = f"{expr_str}"
return {
"Simple_Instruct": f"Instruction: Generate a mathematical expression using variables {variables} and operands {all_ops} and {constants} as constant.\nExpression: {wrapped_expr}",
"Key_Value": f"Variables: {variables}\nOperands: {all_ops}\nConstant: {constants}\nExpression: {wrapped_expr}",
"Delimiter_Based": f"Input: Variables={variables}, Operands={all_ops}, Constant={constants}\nOutput: {wrapped_expr}",
"Minimalist": f"{variables} | {all_ops} | {constants} => {wrapped_expr}"
}
def generate_expression_instruction(expr_str):
max_var = infer_max_var(expr_str)
variables = [f"x_{i}" for i in range(1, max_var + random.randint(1, (max_var) + 1))]
used_ops = extract_operators(expr_str)
extra_ops = list(set(ALL_OPERANDS) - set(used_ops))
added_ops = random.sample(extra_ops, random.randint(1, len(extra_ops))) if extra_ops else []
all_ops = sorted(set(used_ops + added_ops))
constants = ['C']
wrapped_expr = f"{expr_str}"
return {
#"instriction": f"{','.join(variables)}\n{', '.join(all_ops)}\n{', '.join(constants)}\n{wrapped_expr}"
"instriction": f"vars: {', '.join(variables)}\noper: {', '.join(all_ops)}\ncons: {', '.join(constants)}\nexpr: {wrapped_expr}"
}
#print(generate_expression_instruction("x_1 - (x_4 - C)*(x_3 + exp(C*x_2) + C)")) |