(df, ydim="test_duration")
| 145 | plt.show() |
| 146 | |
| 147 | def plot_comp(df, ydim="test_duration"): |
| 148 | import matplotlib.pyplot as plt |
| 149 | fig, ax = plt.subplots() |
| 150 | |
| 151 | pes = df.pe.unique() |
| 152 | for pe in pes: |
| 153 | df_pe = df.loc[df.pe == pe] |
| 154 | |
| 155 | df_ = df_pe.set_index("poly")[ydim].sort_index() |
| 156 | ax.plot(df_.index, df_.values, "o-") |
| 157 | |
| 158 | ax.legend(pes) |
| 159 | ax.set_xlabel("legendre polynomials") |
| 160 | ax.set_ylabel(ydim) |
| 161 | plt.show() |
| 162 | |
| 163 | def main(): |
| 164 | results_dir = 'results/train/exp_computation' |