|
|
|
|
|
|
|
|
""" |
|
|
Created on Sun Jul 17 06:46:29 PM EDT 2022 |
|
|
author: Ryan Hildebrandt, github.com/ryancahildebrandt |
|
|
""" |
|
|
|
|
|
import numpy as np |
|
|
import pandas as pd |
|
|
import plotly.express as px |
|
|
import random |
|
|
|
|
|
from cluster import * |
|
|
|
|
|
random.seed(42) |
|
|
|
|
|
px.defaults.template = "plotly" |
|
|
|
|
|
def d2_plot(in_df): |
|
|
fig = px.scatter( |
|
|
x = np.array(in_df["d2"].values.tolist())[:,0], |
|
|
y = np.array(in_df["d2"].values.tolist())[:,1], |
|
|
color = list(map(str, in_df["cluster"])), |
|
|
hover_name = in_df["prep"] |
|
|
) |
|
|
fig.update_layout(showlegend=False) |
|
|
return fig |
|
|
|
|
|
def d3_plot(in_df): |
|
|
fig = px.scatter_3d( |
|
|
x = np.array(in_df["d3"].values.tolist())[:,0], |
|
|
y = np.array(in_df["d3"].values.tolist())[:,1], |
|
|
z = np.array(in_df["d3"].values.tolist())[:,2], |
|
|
color = list(map(str, in_df["cluster"])), |
|
|
hover_name = in_df["prep"] |
|
|
) |
|
|
fig.update_traces(marker={'size': 2}) |
|
|
fig.update_layout(showlegend=False) |
|
|
return fig |
|
|
|
|
|
def viz_ex(func, cldata, use_embs): |
|
|
dim_df = pd.DataFrame({ |
|
|
"prep" : cldata, |
|
|
"cluster" : cluster_hdbscan(use_embs, 1.0, "euclidean", 5), |
|
|
"emb" : list(use_embs), |
|
|
"d2" : list(func[0]), |
|
|
"d3" : list(func[1]) |
|
|
}) |
|
|
out = {"d2": d2_plot(dim_df), "d3":d3_plot(dim_df)} |
|
|
return out |
