针对最大深度的图决策树训练/测试精度

Question

我试图从决策树模型中绘制出火车和测试集的准确性。由于我不熟悉python，因此我不确定应该使用哪种类型的图形包。我使用了一个简单的for循环来获取打印结果，但不确定[我如何绘制它]。

谢谢！

我的代码：

for x in max_depth_list :

  dtc =DecisionTreeClassifier(max_depth=x)
  dtc.fit(train_x,train_y)

  train_z = dtc.predict(train_x)
  train_z_prob = dtc.predict_proba(train_x)[:,1]

  test_z = dtc.predict(test_x)
  test_z_prob = dtc.predict_proba(test_x)[:,1]

  print("split: {}".format(x))
  print("model accuracy: {}".format(accuracy_score(test_y, test_z)))
所需图enter image description here

我试图从决策树模型中绘制出火车和测试集的准确性。由于我不熟悉python，因此我不确定应该使用哪种类型的图形包。我用了一个简单的...

Answer 1

您发布的图像中的图很可能是使用matplotlib.pyplot模块创建的。假设您已导入其他必要的依赖项，则可以通过执行类似的操作来绘制类似的图形：