ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix poltly plots converting NaN to 'nan' instead of null (issue #373)

Browse Source
This commit is contained in:
allegroai 2021-08-20 00:30:11 +03:00
parent 99327ecb62
commit 8e31e789b4
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
clearml/backend_interface/metrics/reporter.py
View File

@ -283,7 +283,7 @@ class Reporter(InterfaceBase, AbstractContextManager, SetupUploadMixin, AsyncMan
logger=logger, logger=logger,
) )
def report_plot(self, title, series, plot, iter, round_digits=None): def report_plot(self, title, series, plot, iter, round_digits=None, nan_as_null=True):
""" """
Report a Plotly chart Report a Plotly chart
:param title: Title (AKA metric) :param title: Title (AKA metric)
@ -295,13 +295,15 @@ class Reporter(InterfaceBase, AbstractContextManager, SetupUploadMixin, AsyncMan
:param iter: Iteration number :param iter: Iteration number
:param round_digits: number of digits after the dot to leave :param round_digits: number of digits after the dot to leave
:type round_digits: int :type round_digits: int
:param nan_as_null: If True, Convert NaN to None (null), otherwise use 'nan'
:type nan_as_null: bool
""" """
inf_value = math.inf if six.PY3 else float("inf") inf_value = math.inf if six.PY3 else float("inf")
def to_base_type(o): def to_base_type(o):
if isinstance(o, float): if isinstance(o, float):
if o != o: if o != o:
return 'nan' return None if nan_as_null else 'nan'
elif o == inf_value: elif o == inf_value:
return 'inf' return 'inf'
elif o == -inf_value: elif o == -inf_value:
@ -501,6 +503,7 @@ class Reporter(InterfaceBase, AbstractContextManager, SetupUploadMixin, AsyncMan
series=self._normalize_name(series), series=self._normalize_name(series),
plot=plotly_dict, plot=plotly_dict,
iter=iter, iter=iter,
nan_as_null=False,
) )
def report_table(self, title, series, table, iteration, layout_config=None): def report_table(self, title, series, table, iteration, layout_config=None):
@ -525,6 +528,7 @@ class Reporter(InterfaceBase, AbstractContextManager, SetupUploadMixin, AsyncMan
plot=table_output, plot=table_output,
iter=iteration, iter=iteration,
round_digits=False, round_digits=False,
nan_as_null=False,
) )
def report_line_plot(self, title, series, iter, xtitle, ytitle, mode='lines', reverse_xaxis=False, def report_line_plot(self, title, series, iter, xtitle, ytitle, mode='lines', reverse_xaxis=False,
@ -568,6 +572,7 @@ class Reporter(InterfaceBase, AbstractContextManager, SetupUploadMixin, AsyncMan
series='', series='',
plot=plotly_dict, plot=plotly_dict,
iter=iter, iter=iter,
nan_as_null=False,
) )
def report_2d_scatter(self, title, series, data, iter, mode='lines', xtitle=None, ytitle=None, labels=None, def report_2d_scatter(self, title, series, data, iter, mode='lines', xtitle=None, ytitle=None, labels=None,
@ -609,6 +614,7 @@ class Reporter(InterfaceBase, AbstractContextManager, SetupUploadMixin, AsyncMan
series=self._normalize_name(series), series=self._normalize_name(series),
plot=plotly_dict, plot=plotly_dict,
iter=iter, iter=iter,
nan_as_null=False,
) )
def report_3d_scatter(self, title, series, data, iter, labels=None, mode='lines', color=((217, 217, 217, 0.14),), def report_3d_scatter(self, title, series, data, iter, labels=None, mode='lines', color=((217, 217, 217, 0.14),),
@ -679,6 +685,7 @@ class Reporter(InterfaceBase, AbstractContextManager, SetupUploadMixin, AsyncMan
series=self._normalize_name(series) if not isinstance(series, list) else None, series=self._normalize_name(series) if not isinstance(series, list) else None,
plot=plotly_obj, plot=plotly_obj,
iter=iter, iter=iter,
nan_as_null=False,
) )
def report_value_matrix(self, title, series, data, iter, xtitle=None, ytitle=None, xlabels=None, ylabels=None, def report_value_matrix(self, title, series, data, iter, xtitle=None, ytitle=None, xlabels=None, ylabels=None,
@ -722,6 +729,7 @@ class Reporter(InterfaceBase, AbstractContextManager, SetupUploadMixin, AsyncMan
series=self._normalize_name(series), series=self._normalize_name(series),
plot=plotly_dict, plot=plotly_dict,
iter=iter, iter=iter,
nan_as_null=False,
) )
def report_value_surface(self, title, series, data, iter, xlabels=None, ylabels=None, def report_value_surface(self, title, series, data, iter, xlabels=None, ylabels=None,
@ -767,6 +775,7 @@ class Reporter(InterfaceBase, AbstractContextManager, SetupUploadMixin, AsyncMan
series=self._normalize_name(series), series=self._normalize_name(series),
plot=plotly_dict, plot=plotly_dict,
iter=iter, iter=iter,
nan_as_null=False,
) )
def report_image_plot_and_upload(self, title, series, iter, path=None, matrix=None, def report_image_plot_and_upload(self, title, series, iter, path=None, matrix=None,
@ -835,6 +844,7 @@ class Reporter(InterfaceBase, AbstractContextManager, SetupUploadMixin, AsyncMan
series=self._normalize_name(series), series=self._normalize_name(series),
plot=plotly_dict, plot=plotly_dict,
iter=iter, iter=iter,
nan_as_null=False,
) )
def report_console(self, message, level=logging.INFO): def report_console(self, message, level=logging.INFO):