zchen/b2txt25
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

additional documentation

Browse Source
This commit is contained in:
nckcard
2025-07-02 14:28:34 -07:00
parent b2d25bd606
commit d87462df51
6 changed files with 48 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
model_training/evaluate_model.py
View File

@@ -262,4 +262,9 @@ if eval_type == 'val':
output_file = os.path.join(model_path, f'baseline_rnn_{eval_type}_predicted_sentences_{time.strftime("%Y%m%d_%H%M%S")}.txt')
with open(output_file, 'w') as f:
for i in range(len(lm_results['pred_sentence'])):
f.write(f"{remove_punctuation(lm_results['pred_sentence'][i])}\n")
if i < len(lm_results['pred_sentence']) - 1:
# write sentence + newline
f.write(f"{remove_punctuation(lm_results['pred_sentence'][i])}\n")
else:
# don't add a newline at the end of the last sentence
f.write(f"{remove_punctuation(lm_results['pred_sentence'][i])}")