Update japanese-task-evaluation.py

wandb · Nov 1, 2023 · 74842ba · 74842ba
1 parent d26e7bd
commit 74842ba
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/japanese-task-evaluation.py b/src/japanese-task-evaluation.py
@@ -100,7 +100,8 @@
         llm = HuggingFacePipeline(pipeline=pipe)
         llm_chain = LLMChain(llm=llm, prompt=get_template(eval_category[2], template_type), output_key="output")
         jnli_score, jnli_balanced_score = eval_JNLI(dataset,llm_chain)
-        table_contents.append(jnli_score, jnli_balanced_score)
+        table_contents.append(jnli_score)
+        table_contents.append(jnli_balanced_score)
 
         #JSQuAD--------------------------------------------------------
         if config.use_artifact: