MikeMann commited on
Commit
a4dc619
·
1 Parent(s): 908526b

added EvalDataset Generation

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -684,7 +684,7 @@ class BSIChatbot:
684
 
685
  def generateEvalDataset(self):
686
  filepath = "/home/user/app/docs/_eval/BSI_Lektion_Ground_Truth.CSV"
687
- with open(filepath, mode='r') as file:
688
  # Create a CSV reader with DictReader
689
  csv_reader = csv.DictReader(file, delimiter="|")
690
 
@@ -698,15 +698,19 @@ class BSIChatbot:
698
 
699
  # Print the list of dictionaries
700
  for data in data_list:
701
- print(data)
702
  data["Context"] = None
703
  data["Answer"] = None
704
  print(data)
705
-
706
- for data in data_list[:3]:
 
 
 
707
  #def ragPromptNew(self, query, rerankingStep, history, stepBackPrompt)
708
  data["Answer"],data["Context"] = ragPromptNew(data["Frage"],True,None,True)
709
  print(data)
 
710
 
711
  with open('/home/user/app/docs/_eval/eval_dataset.json', 'w') as fout:
712
  json.dump(data_list, fout)
 
684
 
685
  def generateEvalDataset(self):
686
  filepath = "/home/user/app/docs/_eval/BSI_Lektion_Ground_Truth.CSV"
687
+ with open(filepath, mode='r', encoding="latin1", errors="replace") as file:
688
  # Create a CSV reader with DictReader
689
  csv_reader = csv.DictReader(file, delimiter="|")
690
 
 
698
 
699
  # Print the list of dictionaries
700
  for data in data_list:
701
+ print("starting to generate evaldataset..")
702
  data["Context"] = None
703
  data["Answer"] = None
704
  print(data)
705
+ i=1
706
+ #for data in data_list[:3]:
707
+ for data in data_list:
708
+ print("Eval Entry no:")
709
+ print(i)
710
  #def ragPromptNew(self, query, rerankingStep, history, stepBackPrompt)
711
  data["Answer"],data["Context"] = ragPromptNew(data["Frage"],True,None,True)
712
  print(data)
713
+ i=i+1
714
 
715
  with open('/home/user/app/docs/_eval/eval_dataset.json', 'w') as fout:
716
  json.dump(data_list, fout)