ValueError: Unknown label type: \’continuous’错误解决

今天在做决策树鸢尾花分类时出现了一个错误：

解决方法是：train_y后加上astype(‘int’)即可：