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

Add support for datetime in scatter plots + matplotlib support

Browse Source
This commit is contained in:
allegroai 2021-01-24 09:13:32 +02:00
parent dd03ca0d1d
commit 6b8f21b30e
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
clearml/binding/matplotlib_bind.py
View File

@ -372,10 +372,13 @@ class PatchedMatplotlib:
except Exception:
# noinspection PyBroadException
try:
plotly_renderer.plotly_fig['layout']['xaxis1'].update({
_xaxis = next(x for x in ('xaxis', 'xaxis0', 'xaxis1')
if x in plotly_renderer.plotly_fig['layout'])
plotly_renderer.plotly_fig['layout'][_xaxis].update({
'ticktext': [t.get_text() for t in x_ticks],
'tickvals': [t.get_position()[0] for t in x_ticks],
})
plotly_renderer.plotly_fig['layout'][_xaxis].pop('type', None)
except Exception:
pass
y_ticks = list(plotly_renderer.current_mpl_ax.get_yticklabels())
@ -387,10 +390,13 @@ class PatchedMatplotlib:
except Exception:
# noinspection PyBroadException
try:
plotly_renderer.plotly_fig['layout']['yaxis1'].update({
_yaxis = next(x for x in ('yaxis', 'yaxis0', 'yaxis1')
if x in plotly_renderer.plotly_fig['layout'])
plotly_renderer.plotly_fig['layout']['_yaxis'].update({
'ticktext': [t.get_text() for t in y_ticks],
'tickvals': [t.get_position()[1] for t in y_ticks],
})
plotly_renderer.plotly_fig['layout'][_yaxis].pop('type', None)
except Exception:
pass
return deepcopy(plotly_renderer.plotly_fig)

2
clearml/logger.py
View File

@ -485,7 +485,7 @@ class Logger(object):
return self._task._reporter.report_2d_scatter(
title=title,
series=series,
data=scatter.astype(np.float32),
data=scatter,
iter=iteration,
mode=mode,
xtitle=xaxis,