[linux] matplotlib plt画training dynamics指标曲线时，标记每个点的值

最新推荐文章于 2024-06-10 10:32:50 发布

心心喵

最新推荐文章于 2024-06-10 10:32:50 发布

阅读量393

点赞数 10

分类专栏： linux 文章标签： python 机器学习开发语言

本文链接：https://blog.csdn.net/Trance95/article/details/136446742

版权

linux 专栏收录该内容

148 篇文章 3 订阅

订阅专栏

该文章详细介绍了如何在Python的matplotlib库中，通过plot_ret和plot_ret_8M函数绘制折线图，并使用plt.annotate标记折线图上的数值点。这两个函数分别处理不同数据集，展示了如何设置标签、坐标轴和图例，以及网格线和x轴刻度。

摘要由CSDN通过智能技术生成

plt画折线图时，plt.annotate标记折线图的点的数值。

def plot_ret(*ret_dicts):
    plt.figure(figsize=(10, 5))
    for ret_dict in ret_dicts:
        print(ret_dict["iters"])
        plt.plot([iter*4/1000 for iter in ret_dict["iters"]], ret_dict["ret"], marker='o', markersize=5, markerfacecolor='white', label=ret_dict["model"][:60]) #label=ret_dict["model"][:60]其中label比较长的需要取到所有字符，所以从24改为60
        # for i, txt in enumerate(ret_dict["ret"]):
            # plt.annotate("{:.3f}".format(txt), ([iter*4/1000 for iter in ret_dict["iters"]][i], ret_dict["ret"][i]))
        title = ret_dict["benchmark"]
    ax = plt.gca()
    ax.xaxis.set_major_locator(ticker.MultipleLocator(20))
    plt.legend(prop={"size":6})
    plt.grid(True)
    plt.title(title)
    plt.xlabel("token(B)")
    # plt.show()

def plot_ret_8M(*ret_dicts):
    # plt.figure(figsize=(10, 5))
    for ret_dict in ret_dicts:
        print(ret_dict["iters"])
        plt.plot([iter*8/1000 for iter in ret_dict["iters"]], ret_dict["ret"], marker='o', markersize=5, markerfacecolor='white', label=ret_dict["model"][:60])
        for i, txt in enumerate(ret_dict["ret"]):
            plt.annotate("{:.3f}".format(txt), ([iter*8/1000 for iter in ret_dict["iters"]][i], ret_dict["ret"][i]))
        title = ret_dict["benchmark"]
    ax = plt.gca()
    ax.xaxis.set_major_locator(ticker.MultipleLocator(20))
    plt.legend(prop={"size":6})
    plt.grid(True)
    plt.title(title)
    plt.xlabel("token(B)")
    plt.show()