add visu code
def cluster_epochs