-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy paththread.yaml
52 lines (52 loc) · 1.87 KB
/
thread.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
output_dir: "thread_results" # Set your output directory in case you dont log to console
store: True # Set to True if you want to store the results in a file in the output directory (see above)
seed: 42
task: "THREAD" # "GENSTYLETHREAD" to generate online profiles and save to json
dryrun: False # Not relevant
save_prompts: True # Not relevant
timeout: 0.0 # Not relevant
gen_model:
name: "gpt-4-1106-preview"
provider: "openai"
args: {
temperature: 1.
}
task_config:
no_threads: 1
no_rounds: 2 # optimal: 2-3
no_actions: 3 # optimal: 2-3
no_max_comments: 3
max_depth: 5 # optimal: 5-6
no_profiles: 40
p_critic: 0.4 # optimal: 0.4
p_short: 0.7
mode: "top" # used top mode for dataset; random is not effective
no_sampled_comments: 10 # optimal: 5-15
default_comment_prob: 7 # optimal: 7
min_comment_len: 1
max_comment_len: 10
guess_feature: ["city_country", "age", "income_level", "income", "education", "occupation", "sex", "relationship_status", "birth_city_country"] # provide as a list
author_bot_system_prompt_path: "./data/thread/system_prompts/author_system_prompt.txt"
user_bot_system_prompt_path: "./data/thread/system_prompts/user_system_prompt.txt"
profile_checker_prompt_path: "./data/thread/system_prompts/profile_checker_prompt.txt"
user_style_prompt_path: "./data/thread/system_prompts/user_style_system_prompt.txt"
user_bot_personalities_path: "./data/profiles/user_bot_gen_online_profiles_300.json"
author_bot:
name: "gpt-4-1106-preview"
provider: "openai"
args: {
temperature: 1.
}
user_bot:
name: "gpt-4-1106-preview"
provider: "openai"
args: {
temperature: 1.,
frequency_penalty: 2.0
}
checker_bot:
name: "gpt-4-1106-preview"
provider: "openai"
args: {
temperature: 0.
}