How to visualize the results?

[easternbun]

How to visualize the results like cluster group graph and image output with different group.

[agadetsky]

Dear @easternbun,

Thank you for your interest in our work. First, you can extract the cluster indices for each sample in the dataset similarly to evaluate.py script

turtle/evaluate.py

Line 44 in 9b8bbb7

y_pred = labels.argmax(dim=-1).detach().cpu().numpy()

Then, you can use your favourite library to do visualizations.

Let me know if that has resolved your issue.

Best,
Artyom