ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-23 01:55:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Logger.report_table() support for table as list of lists

Browse Source
This commit is contained in:
allegroai 2020-11-01 16:01:24 +02:00
parent 236e6adcfd
commit b8ebbaa3b8
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
trains/logger.py
View File

@ -265,7 +265,7 @@ class Logger(object):
title, # type: str title, # type: str
series, # type: str series, # type: str
iteration, # type: int iteration, # type: int
table_plot=None, # type: Optional[pd.DataFrame] table_plot=None, # type: Optional[pd.DataFrame, Sequence[Sequence]]
csv=None, # type: Optional[str] csv=None, # type: Optional[str]
url=None, # type: Optional[str] url=None, # type: Optional[str]
extra_layout=None, # type: Optional[dict] extra_layout=None, # type: Optional[dict]
@ -275,7 +275,7 @@ class Logger(object):
One and only one of the following parameters must be provided. One and only one of the following parameters must be provided.
- ``table_plot`` - Pandas DataFrame - ``table_plot`` - Pandas DataFrame or Table as list of rows (list)
- ``csv`` - CSV file - ``csv`` - CSV file
- ``url`` - URL to CSV file - ``url`` - URL to CSV file
@ -296,7 +296,7 @@ class Logger(object):
:param str series: The series name (variant) of the reported table. :param str series: The series name (variant) of the reported table.
:param int iteration: The iteration number. :param int iteration: The iteration number.
:param table_plot: The output table plot object :param table_plot: The output table plot object
:type table_plot: pandas.DataFrame :type table_plot: pandas.DataFrame or Table as list of rows (list)
:param csv: path to local csv file :param csv: path to local csv file
:type csv: str :type csv: str
:param url: A URL to the location of csv file. :param url: A URL to the location of csv file.
@ -325,10 +325,13 @@ class Logger(object):
for src in srcs: for src in srcs:
reporter_table.replace(src, dst, inplace=True) reporter_table.replace(src, dst, inplace=True)
reporter_table = table.fillna(str(np.nan)) if isinstance(table, (list, tuple)):
replace("NaN", np.nan, math.nan) reporter_table = table
replace("Inf", np.inf, math.inf) else:
replace("-Inf", -np.inf, np.NINF, -math.inf) reporter_table = table.fillna(str(np.nan))
replace("NaN", np.nan, math.nan)
replace("Inf", np.inf, math.inf)
replace("-Inf", -np.inf, np.NINF, -math.inf)
# noinspection PyProtectedMember # noinspection PyProtectedMember
return self._task._reporter.report_table( return self._task._reporter.report_table(
title=title, title=title,