mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-07-02 11:24:56 +00:00
20 lines
454 B
Python
20 lines
454 B
Python
|
import os
|
||
|
|
import seaborn as sns
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
def plot_metric(train, val, name, dir_output):
|
||
|
|
plt.plot(train, label=f"train_{name}")
|
||
|
|
plt.plot(val, label=f"val_{name}")
|
||
|
|
plt.title(name)
|
||
|
|
plt.xlabel("epoch")
|
||
|
|
plt.ylabel(name)
|
||
|
|
plt.savefig(os.path.join(dir_output, f"{name}.png"))
|
||
|
|
plt.close()
|
||
|
|
|
||
|
|
FILEPATH_INPUT = os.environ["INPUT"]
|
||
|
|
DIRPATH_OUTPUT = os.environ["OUTPUT"] if "OUTPUT" in os.environ else os.getcwd()
|
||
|
|
|
||
|
|
|
||
|
|
df = pd.read_csv(FILEPATH_INPUT)
|
||
|
|
|