changes in comparing results

This commit is contained in:
Godmode Galactus 2023-06-09 11:32:10 +02:00
parent 52886c1a6b
commit 683db0dd49
No known key found for this signature in database
GPG Key ID: A04142C71ABB0DEA
1 changed files with 14 additions and 3 deletions

View File

@ -7,6 +7,7 @@ import matplotlib.pyplot as plt
path_1 = sys.argv[1]
path_2 = sys.argv[2]
path_3 = sys.argv[3]
def extract_data(path):
json_files = [pos_json for pos_json in os.listdir(path) if pos_json.endswith('.json')]
@ -32,6 +33,7 @@ def get_metrics(key, runs, plot):
runs_1 = extract_data(path_1)
runs_2 = extract_data(path_2)
runs_3 = extract_data(path_3)
keysA = get_keys(runs_1)
keysB = get_keys(runs_2)
@ -45,43 +47,52 @@ for key in keysA:
figure, axis = plt.subplots(3, 3)
[data_keys_1, metrics_1] = get_metrics(key, runs_1, plot)
[data_keys_2, metrics_2] = get_metrics(key, runs_2, plot)
[data_keys_3, metrics_3] = get_metrics(key, runs_3, plot)
axis[0, 0].set_title(plot[0], fontsize=5)
l1 = axis[0, 0].plot(data_keys_1, metrics_1[0], 'r', label=path_1)
l2 = axis[0, 0].plot(data_keys_2, metrics_2[0], 'b', label=path_2)
axis[0, 0].plot(data_keys_1, metrics_1[0], 'r', label=path_1)
axis[0, 0].plot(data_keys_2, metrics_2[0], 'b', label=path_2)
axis[0, 0].plot(data_keys_3, metrics_3[0], 'g', label=path_3)
axis[0, 1].set_title(plot[1], fontsize=5)
axis[0, 1].plot(data_keys_1, metrics_1[1], 'r')
axis[0, 1].plot(data_keys_2, metrics_2[1], 'b')
axis[0, 1].plot(data_keys_3, metrics_3[1], 'g')
axis[0, 2].set_title(plot[2], fontsize=5)
axis[0, 2].plot(data_keys_1, metrics_1[2], 'r')
axis[0, 2].plot(data_keys_2, metrics_2[2], 'b')
axis[0, 2].plot(data_keys_3, metrics_3[2], 'g')
axis[1, 0].set_title(plot[3], fontsize=5)
axis[1, 0].plot(data_keys_1, metrics_1[3], 'r')
axis[1, 0].plot(data_keys_2, metrics_2[3], 'b')
axis[1, 0].plot(data_keys_3, metrics_3[3], 'g')
axis[1, 1].set_title(plot[4], fontsize=5)
axis[1, 1].plot(data_keys_1, metrics_1[4], 'r')
axis[1, 1].plot(data_keys_2, metrics_2[4], 'b')
axis[1, 1].plot(data_keys_3, metrics_3[4], 'g')
axis[1, 2].set_title(plot[5], fontsize=5)
axis[1, 2].plot(data_keys_1, metrics_1[5], 'r')
axis[1, 2].plot(data_keys_2, metrics_2[5], 'b')
axis[1, 2].plot(data_keys_3, metrics_3[5], 'g')
axis[2, 0].set_title(plot[6], fontsize=5)
axis[2, 0].plot(data_keys_1, metrics_1[6], 'r')
axis[2, 0].plot(data_keys_2, metrics_2[6], 'b')
axis[2, 0].plot(data_keys_3, metrics_3[6], 'g')
axis[2, 1].set_title(plot[7], fontsize=5)
axis[2, 1].plot(data_keys_1, metrics_1[7], 'r')
axis[2, 1].plot(data_keys_2, metrics_2[7], 'b')
axis[2, 1].plot(data_keys_3, metrics_3[7], 'g')
handles, labels = axis[0, 0].get_legend_handles_labels()
figure.legend(handles, labels, loc='upper center')
figure.legend(handles, labels, loc='upper left')
figure.tight_layout()
plt.savefig(key+'.png', dpi=1200)