File size: 1,179 Bytes
1b837e6 1fc5e1f 1b837e6 1fc5e1f 1b837e6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
from renumics import spotlight
from renumics.spotlight.analysis.typing import DataIssue
import pandas as pd
import numpy as np
if __name__ == "__main__":
df = pd.read_json("sd_dataset.json")
data_issues = sorted(df["selection"].unique())
spotlight_data_issues = []
for issue in data_issues:
if issue == -1:
continue
issue_rows = np.where(df["selection"] == issue)[0].tolist()
new_issue = DataIssue(
severity="medium",
title=f"Best Images Cluster {issue}",
description=f"Best Images Cluster {issue}",
rows=issue_rows)
spotlight_data_issues.append(new_issue)
while True:
dtypes = {
"image": spotlight.Image,
"clip_image_embedding": spotlight.Embedding,
"clip_text_embedding": spotlight.Embedding,
}
view = spotlight.show(
df,
dtype=dtypes,
layout="spotlight-layout.json",
issues=spotlight_data_issues,
port=7860,
host="0.0.0.0",
allow_filebrowsing=False
)
view.close()
|