__init__.py 文件源码

def print_graph_test(y_test, y_pred1, y_pred2, maxEntries=50):
    #y_pred_limit = min(maxEntries, len(y_test))
    length = min(maxEntries,len(y_test))
    y_pred_limit = np.linspace(1, length, length)
    plt.plot(y_pred_limit, y_test, label='actual', color='b', linewidth=1)
    plt.plot(y_pred_limit, y_pred1, '--', color='r', linewidth=2, label='prediction1')
    plt.plot(y_pred_limit, y_pred2, '--', color='g', linewidth=2, label='prediction2')
    plt.legend(loc=0)
    plt.show()