Python script to read final affinity matrix and overlay cluster assignments.