correlation.py 文件源码

def poly_fit(x, y, degree):
    results = {}
    co_effs = np.polyfit(x, y, degree)

    results['poly_normal'] = co_effs

    p = np.poly1d(co_effs)

    y_hat = p(x)
    y_bar = np.sum(y) / len(y)

    ss_reg = np.sum((y_hat - y_bar) ** 2)
    ss_tot = np.sum((y - y_bar) ** 2)
    results['determination'] = ss_reg / ss_tot

    return results


# noinspection PyTypeChecker