mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-16 22:53:00 +00:00
46 lines
1.2 KiB
Python
Executable file
46 lines
1.2 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import seaborn as sns
|
|
import matplotlib.pyplot as plt
|
|
import pandas as pd
|
|
|
|
def plot_metric(ax, train, val, name, dir_output):
|
|
ax.plot(train, label=f"train_{name}")
|
|
ax.plot(val, label=f"val_{name}")
|
|
ax.set_title(name)
|
|
ax.set_xlabel("epoch")
|
|
ax.set_ylabel(name)
|
|
# plt.savefig(os.path.join(dir_output, f"{name}.png"))
|
|
# plt.close()
|
|
|
|
|
|
def plot_metrics(filepath_input, dirpath_output):
|
|
df = pd.read_csv(filepath_input, sep="\t")
|
|
|
|
fig = plt.figure(figsize=(10,13))
|
|
for i, colname in enumerate(filter(lambda colname: colname != "epoch" and not colname.startswith("val_"), df.columns.values.tolist())):
|
|
train = df[colname]
|
|
val = df[f"val_{colname}"]
|
|
|
|
colname_display = colname.replace("metric_dice_coefficient", "dice coefficient") \
|
|
.replace("one_hot_mean_iou", "mean iou")
|
|
|
|
ax = fig.add_subplot(3, 2, i+1)
|
|
|
|
plot_metric(ax, train, val, name=colname_display, dir_output=dirpath_output)
|
|
|
|
fig.tight_layout()
|
|
|
|
target=os.path.join(dirpath_output, f"metrics.png")
|
|
plt.savefig(target)
|
|
|
|
print(f">>> Saved to {target}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
FILEPATH_INPUT = os.environ["INPUT"]
|
|
DIRPATH_OUTPUT = os.environ["OUTPUT"] if "OUTPUT" in os.environ else os.getcwd()
|
|
|
|
plot_metrics(FILEPATH_INPUT, DIRPATH_OUTPUT)
|
|
|