the print “Probability of belongs to cluster 1:” should be “Probability of belongs to cluster 0:” ?? or i miss the meanings?