| import os |
| import sys |
| import itertools |
|
|
|
|
| def main(): |
| input_folder = 'data_ToMh' |
| output_folder = 'prompt_ToMh' |
| lengths = [1, 2, 3] |
| orders = [0, 1, 2, 3, 4] |
| prompts = ['CoT', 'MC'] |
| tells = ['No_Tell', 'Tell'] |
| for tell, prompt, length, order, sample_num in itertools.product(tells, prompts, lengths, orders, range(1, 21)): |
| input_fn = os.path.join(input_folder, tell, prompt, f'length_{length}', f'sample_{sample_num}', |
| f'order_{order}.txt') |
| output_fn = os.path.join(output_folder, tell, prompt, f'length_{length}', f'sample_{sample_num}', |
| f'order_{order}.txt') |
| with open(input_fn, 'r') as file: |
| lines = file.readlines() |
| new_lines = [line for line in lines if line == |
| '\n' or line.split()[0] != 'Answer:'] |
| if not os.path.exists(os.path.join(output_folder, tell, prompt, f'length_{length}', f'sample_{sample_num}')): |
| os.makedirs(os.path.join(output_folder, tell, prompt, |
| f'length_{length}', f'sample_{sample_num}')) |
| with open(output_fn, 'w') as file: |
| file.writelines(new_lines) |
|
|
|
|
|
|
| if __name__ == "__main__": |
| sys.exit(main()) |
|
|
