namberino commited on
Commit
6cc69fa
·
1 Parent(s): 611d166

Update questions per page

Browse files
Files changed (1) hide show
  1. generator.py +2 -2
generator.py CHANGED
@@ -1093,7 +1093,7 @@ class RAGMCQWithDifficulty(RAGMCQ):
1093
  pdf_path: str,
1094
  n_questions: int = 10,
1095
  mode: str = "rag", # per_page or rag
1096
- questions_per_page: int = 3, # for per_page mode
1097
  top_k: int = 3, # chunks to retrieve for each question in rag mode
1098
  temperature: float = 0.2,
1099
  enable_fiddler: bool = False,
@@ -1217,7 +1217,7 @@ class RAGMCQWithDifficulty(RAGMCQ):
1217
  collection: str,
1218
  n_questions: int = 10,
1219
  mode: str = "rag", # 'per_chunk' or 'rag'
1220
- questions_per_chunk: int = 3, # used for 'per_chunk'
1221
  top_k: int = 3, # retrieval size used in RAG
1222
  temperature: float = 0.2,
1223
  enable_fiddler: bool = False,
 
1093
  pdf_path: str,
1094
  n_questions: int = 10,
1095
  mode: str = "rag", # per_page or rag
1096
+ questions_per_page: int = 5, # for per_page mode
1097
  top_k: int = 3, # chunks to retrieve for each question in rag mode
1098
  temperature: float = 0.2,
1099
  enable_fiddler: bool = False,
 
1217
  collection: str,
1218
  n_questions: int = 10,
1219
  mode: str = "rag", # 'per_chunk' or 'rag'
1220
+ questions_per_chunk: int = 5, # used for 'per_chunk'
1221
  top_k: int = 3, # retrieval size used in RAG
1222
  temperature: float = 0.2,
1223
  enable_fiddler: bool = False,