Update srbench_gen.py

This commit is contained in:
Jun Yao 2025-05-30 11:23:26 +08:00 committed by GitHub
parent f038ffac17
commit ae2d1fea62
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -13,7 +13,7 @@ INFER_TEMPLATE = f'''
Based on the above data, please infer the possible formula. Ensure that your inference applies to all the provided data points, and consider both linear and nonlinear combinations.
Verify whether your formula applies to the following new data point and adjust it to ensure accuracy:
{{prompt2}}
Finally, please output only the formula string you inferred (e.g. z=x_0 * x_1), without any additional information.
Finally, please output only the formula string you inferred (e.g. y=x_0 * x_1), without any additional information.
'''
srbench_reader_cfg = dict(input_columns=["prompt1","prompt2"], output_column='Formula')