Skip to content
Snippets Groups Projects
Commit eec7e265 authored by yilun_jin's avatar yilun_jin
Browse files

Update local_evaluation.py

parent 467330ad
No related branches found
No related tags found
No related merge requests found
......@@ -57,6 +57,8 @@ def generate_model_outputs(data_df, model):
data_df.iterrows(), total=len(data_df), desc="Generating Responses"
):
is_multiple_choice = row["task_type"] == "multiple-choice"
# the 'task_type' column won't be available during evaluation, so you should use something like
# ```is_multiple_choice = row['is_multiple_choice']``
prompt = row["input_field"]
model_output = model.predict(prompt, is_multiple_choice)
outputs.append(model_output)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment