fixes for json serialize bug

This commit is contained in:
Mikayla Dobson
2023-01-08 11:20:57 -06:00
parent 8f2d9faf1e
commit b7648dc4af
3 changed files with 20 additions and 26 deletions

View File

@@ -5,13 +5,13 @@ def format_result(app_config: Config, json_path):
# dictionary to hold and later display our results
insertions_by_label = {}
data_path = app_config.data_path
data_path = app_config['data_path']
# if pg_config is not None, run the postgres prediction[0] of this code
pg_config = app_config.pg_config
pg_config = app_config['pg_config']
# if this is True, run the prediction[0] "for line in contents:" below
sort_by_match_strength = app_config.sort_by_match_strength
sort_by_match_strength = app_config['sort_by_match_strength']
weak_results = 0
total_count = 0
@@ -52,6 +52,9 @@ def format_result(app_config: Config, json_path):
if not guess_label in insertions_by_label:
insertions_by_label[guess_label] = 0
print(img_path)
print("./predictions/" + guess_label)
# copy file to appropriate location, depending on if sorting
if sort_by_match_strength:
if (not os.path.exists("./predictions/" + match_strength + guess_label)):