update scripts

src/backend/evaluate_model.py

@@ -81,12 +81,12 @@ class Evaluator:
             # print(envs.DATASET_PATH)
             # print(df.shape)
             # print(df.iloc[-1])
-            self.generated_summaries_df = self.summary_generator.generate_summaries(envs.DATASET_PATH, df_prompt, save_path=f"generation_results/{self.model}.csv")
+            self.generated_summaries_df = self.summary_generator.generate_summaries(envs.DATASET_PATH, df_prompt, save_path=f"./generation_results/{self.model}.csv")
             # exit()
             # avg_summary_len = self.summary_generator.avg_length
             # answer_rate = self.summary_generator.answer_rate
             '''开始评估模型的结果'''
-            self.humanlike = self.eval_model.evaluate_humanlike(self.generated_summaries_df, envs.HUMAN_DATA, f"generation_results/{self.model}.csv")
+            self.humanlike = self.eval_model.evaluate_humanlike(self.generated_summaries_df, envs.HUMAN_DATA, f"./generation_results/{self.model}.csv")
             '''原始指标'''
             # self.hallucination_scores, self.eval_results = self.eval_model.evaluate_hallucination(
                 # self.generated_summaries_df)

src/backend/model_operations.py

@@ -187,7 +187,7 @@ class SummaryGenerator:
                     # print(ID, q_ID, prompt_value)
                     system_prompt = envs.SYSTEM_PROMPT 
                     _user_prompt = prompt_value
-                    for ii in range(10):
+                    for ii in range(1):
                     # user_prompt = f"{envs.USER_PROMPT}\nPassage:\n{_source}"
                         while True:
                             try: