plot.py 文件源码

def plot_tree_data(data, indicies_x, indicies_y, model):
    plt.subplot(3, 1, 1)
    data, indicies_x, indicies_y, model = load_tree_data()
    data_line, = plt.plot(data, color="blue", label="data")
    data_indicies_line, = plt.plot(
        indicies_x,
        indicies_y,
        "o",
        color="green",
        label="fitness predictors"
    )
    model_line, = plt.plot(model, color="red", label="model")
    plt.title("Data and Model Output")
    plt.legend()

    return data_line, data_indicies_line, model_line