ba-thesis/sw/plot_results.py

44 lines
1.0 KiB
Python

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd
import os
from utility import visualization, simulation
def plot_results():
sim_names = [
"96.3965",
"204.3.486",
"204.55.187",
"408.33.844",
"816.1A4.845",
"999.111.3.5543",
"999.111.3.5565",
"PEGReg252x504"
]
deserializer = simulation.SimulationDeSerializer(save_dir="sim_saves", results_dir="sim_results")
data = []
for sim_name in sim_names:
df, metadata = deserializer.read_results(sim_name)
df = df.loc[:, ~df.columns.str.contains('^Unnamed')]
graph_title = sim_name
line_labels = metadata["labels"]
graph_tuple = (graph_title, df, line_labels)
data.append(graph_tuple)
sns.set_theme()
fig = visualization.plot_BERs(title="Bit-Error-Rates of proximal decoder for different codes",
data=data, num_cols=4)
plt.show()
if __name__ == "__main__":
plot_results()