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 Parallel Coordinate plots handling of categorical variables

Browse Source
This commit is contained in:
Alex Burlacu 2023-03-23 16:12:31 +02:00
parent 3817f0ca36
commit f9fa303091
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/automation/optimization.py
View File

@ -1880,7 +1880,13 @@ class HyperParameterOptimizer(object):
except (ValueError, TypeError):
values = list(range(len(col[1:])))
ticks = col[1:]
unique_ticks = list(set(ticks))
d = dict(label=col[0], values=values, tickvals=values, ticktext=ticks)
if len(ticks) != len(unique_ticks): # Mapping duplicate ticktext
ticktext = {key: i for i, key in enumerate(unique_ticks)}
d["values"] = [ticktext[tick] for tick in ticks]
d["tickvals"] = list(range(len(ticktext)))
d["ticktext"] = list(sorted(ticktext, key=ticktext.get))
pcc_dims.append(d)
# report parallel coordinates
plotly_pcc = dict(