def stashskel(name, ids, labels, colors=None, palette=None, **kws):
saveskels(
name,
ids,
labels,
colors=colors,
palette=None,
foldername=FNAME,
save_on=SAVESKELS,
**kws,
)
# %% Extract the antennal lobe clusters
communities = [2, 3]
from src.io import saveskels
for comm in communities:
inds = partition[partition == comm].index
comm_meta = mg.meta.loc[inds]
class_labels = comm_meta["Merge Class"]
colors = np.vectorize(CLASS_COLOR_DICT.get)(class_labels)
saveskels(
f"cluster-{comm}",
inds,
class_labels,
colors,
multiout=False,
palette=None,
foldername=FNAME,
)
saveskels(
f"cluster-{comm}",
inds,
class_labels,
colors,
multiout=True,
palette=None,
foldername=FNAME,
)
# %% [markdown]
def stashskels(name, ids, labels, **kws):
saveskels(name, ids, labels, foldername=FNAME, **kws)
