コード例 #1
0
ファイル: gen_imgs.py プロジェクト: jackru/pyGAM 
def cake_data_in_one():
    X, y = cake()

    gam = LinearGAM(fit_intercept=True)
    gam.gridsearch(X, y)

    XX = gam.generate_X_grid()

    plt.figure()
    plt.plot(gam.partial_dependence(XX))
    plt.title('LinearGAM')
    plt.savefig('imgs/pygam_cake_data.png', dpi=300)
コード例 #2
0
ファイル: conftest.py プロジェクト: maorn/pyGAM 
def cake_X_y():
    # y is real
    # recommend LinearGAM
    return cake(return_X_y=True)