Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
IliaLarchenko
commited on
Commit
·
bb0f942
1
Parent(s):
1e66988
Small fixes for analysis
Browse files- tests/analysis.py +1 -1
- tests/grader.py +1 -1
tests/analysis.py
CHANGED
@@ -125,7 +125,7 @@ def run_evaluation(
|
|
125 |
|
126 |
exp_name = f"{exp_name}_{pd.Timestamp.now().strftime('%Y-%m-%d_%H-%M-%S')}"
|
127 |
os.makedirs(f"records/{exp_name}", exist_ok=True)
|
128 |
-
tasks = [(type_, i, llm_config) for
|
129 |
complete_f = partial(complete_and_grade, exp_name=exp_name, grader_models=grader_models, candidate_model=candidate_model)
|
130 |
|
131 |
with concurrent.futures.ThreadPoolExecutor(max_workers=num_workers) as executor:
|
|
|
125 |
|
126 |
exp_name = f"{exp_name}_{pd.Timestamp.now().strftime('%Y-%m-%d_%H-%M-%S')}"
|
127 |
os.makedirs(f"records/{exp_name}", exist_ok=True)
|
128 |
+
tasks = [(type_, i, llm_config) for i in range(num_attempts) for type_ in interview_types for llm_config in llm_configs]
|
129 |
complete_f = partial(complete_and_grade, exp_name=exp_name, grader_models=grader_models, candidate_model=candidate_model)
|
130 |
|
131 |
with concurrent.futures.ThreadPoolExecutor(max_workers=num_workers) as executor:
|
tests/grader.py
CHANGED
@@ -28,7 +28,7 @@ def grade(json_file_path, model="gpt-4-turbo", suffix=""):
|
|
28 |
{"role": "user", "content": f"Please evaluate the interviewer based on the following data: \n {'\n'.join(interview_summary_list)}"},
|
29 |
]
|
30 |
|
31 |
-
response = client.chat.completions.create(model=model, messages=messages, temperature=
|
32 |
feedback = json.loads(response.choices[0].message.content)
|
33 |
|
34 |
feedback["file_name"] = json_file_path
|
|
|
28 |
{"role": "user", "content": f"Please evaluate the interviewer based on the following data: \n {'\n'.join(interview_summary_list)}"},
|
29 |
]
|
30 |
|
31 |
+
response = client.chat.completions.create(model=model, messages=messages, temperature=0, response_format={"type": "json_object"})
|
32 |
feedback = json.loads(response.choices[0].message.content)
|
33 |
|
34 |
feedback["file_name"] = json_file_path
|