(df, idx, include_answer=True)
| 155 | |
| 156 | |
| 157 | def format_example(df, idx, include_answer=True): |
| 158 | prompt = df.iloc[idx, 0] |
| 159 | k = df.shape[1] - 2 |
| 160 | for j in range(k): |
| 161 | prompt += "\n{}. {}".format(get_choices()[j], df.iloc[idx, j + 1]) |
| 162 | prompt += "\nAnswer:" |
| 163 | if include_answer: |
| 164 | prompt += " {}\n\n".format(df.iloc[idx, k + 1]) |
| 165 | return prompt |
| 166 | |
| 167 | |
| 168 | def gen_prompt(train_df, subject, k=-1): |
no test coverage detected