ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 18:52:12 +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

9
trains/logger.py
View File

@ -265,7 +265,7 @@ class Logger(object):
title, # type: str
series, # type: str
iteration, # type: int
table_plot=None, # type: Optional[pd.DataFrame]
table_plot=None, # type: Optional[pd.DataFrame, Sequence[Sequence]]
csv=None, # type: Optional[str]
url=None, # type: Optional[str]
extra_layout=None, # type: Optional[dict]
@ -275,7 +275,7 @@ class Logger(object):
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
- ``url`` - URL to CSV file
@ -296,7 +296,7 @@ class Logger(object):
:param str series: The series name (variant) of the reported table.
:param int iteration: The iteration number.
: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
:type csv: str
:param url: A URL to the location of csv file.
@ -325,6 +325,9 @@ class Logger(object):
for src in srcs:
reporter_table.replace(src, dst, inplace=True)
if isinstance(table, (list, tuple)):
reporter_table = table
else:
reporter_table = table.fillna(str(np.nan))
replace("NaN", np.nan, math.nan)
replace("Inf", np.inf, math.inf)