Question

我需要绘制精度与参数图（主要任务是通过网格搜索找到最佳参数值）。现在，因为有许多参数组合，我通过它进行迭代;我想过用迭代次数标记每个组合;并绘制精度与迭代次数的关系曲线。

我需要根据参数值的每个组合绘制精度。但我还想提供一个图表的图例，该图表将参数组合值与其各自的迭代次数相对应。这是怎么回事？

    parameters = parameterGrid(paramGrid)
    i = 0
    lst = []
    maxScore = 0


for parameter in parameters:
    meanAcc = 0

    clf = DecisionTreeClassifier(**parameter)
    dTree = clf.fit(xTrain,yTrain)
    score = dTree.score(xTest, yTest)
    if(score > maxScore):
        maxScore = score
        selectedParams = parameter    
    plt.plot(i,score)
    #plotting accuracy against current parameter set 

    l = [i,score]
    lst.append(l)
    legend[i] = parameter
    i += 1



plt.xlabel('X Label')
plt.ylabel('Y Label')
plt.show()

此外，关于如何进行绘图的任何其他建议都非常受欢迎。请帮忙。

Answer 1

如果要循环使用

的参数

iter = 0
for param1 in [1,2,3,4]:
 for param2 in [2,3,4,5]:
  plt.plot(..., label=str(param1)+"_"+str(param2)+"_"+str(iter))
  iter += 1
plt.legend()

如果没有类标签，如何在matplotlib中定义图例？

1 个答案: