sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-12-22 22:25:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

ai vis: make parallel coordinates wider

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-16 18:51:49 +01:00
parent a70794e661
commit a552cc4dad
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
aimodel/src/lib/vis/embeddings.py
View file

@ -12,17 +12,17 @@ def vis_embeddings(filepath_output, features):
).fit(features)
px = 1 / plt.rcParams['figure.dpi'] # matplotlib sizes are in inches :-( :-( :-(
width = 1920
width = 3840
height = 768
plt.rc("font", size=20)
plt.rc("font", family="Ubuntu")
figure = plt.figure(figsize=(width*px, height*px))
figure.add_subplot(1, 2, 1)
figure, axes = plt.subplot_mosaic("ABBBB", figsize=(width*px, height*px))
# figure.add_subplot(1, 2, 1)
# 1: UMAP
umap.plot.points(dimreducer,
ax=figure.get_axes()[0]
ax=axes["A"]
)
plt.title(f"UMAP Dimensionality Reduction", fontsize=20)
@ -34,7 +34,7 @@ def vis_embeddings(filepath_output, features):
pandas.plotting.parallel_coordinates(
dataframe,
"Label",
ax=figure.get_axes()[1],
ax=axes["B"],
use_columns=False,
axvlines=False,
sort_labels=True