使用sklearn决策树图时出现数据错误

Question

所以这个数据错误，我应该修改哪个数据才能让它正常呢，其显示应为str的类型而不是float，但我尝试过去修改数据，predictions和syar3里的数据都试过修改为str，仍无作用

Answer 1

修改class_names=str(np.unique(Y_train))，这里之所以是Y_train而不是Y，是由于模型的根据训练集构建的。