Python BigModelの例

コード例 #1

ファイル: test_model.py プロジェクト: andreiolariu/kaggle-bulldozer

def run_full():
    train_X, train_Y, _ = get_db_data(tags=['train1', 'train2', 'train3'])
    test_X, _, IDs = get_db_data(tags=['test'])
    
    m = BigModel(columns, n_est=300)
    R = m.train_test(train_X, train_Y, test_X)
    R = np.exp(R)
    write_submission("bigmodelv6.csv", R, IDs)

コード例 #2

def run_full():
    train_X, train_Y, _ = get_db_data(tags=['train1', 'train2', 'train3'])
    test_X, _, IDs = get_db_data(tags=['test'])

    m = BigModel(columns, n_est=300)
    R = m.train_test(train_X, train_Y, test_X)
    R = np.exp(R)
    write_submission("bigmodelv6.csv", R, IDs)

コード例 #3

ファイル: test_model.py プロジェクト: andreiolariu/kaggle-bulldozer

def run_test():
    
    #train_X, train_Y, _ = get_db_data(tags=['train1'])
    #test_X, test_Y, _ = get_db_data(tags=['train2'])
    train_X, train_Y, _ = get_db_data(tags=['train1', 'train2'])
    test_X, test_Y, _ = get_db_data(tags=['train3'])
    
    model = BigModel(columns, n_est=500)
    R2 = model.train_test(train_X, train_Y, test_X)
    print get_rmse(test_Y, R2)

コード例 #4

def run_test():

    #train_X, train_Y, _ = get_db_data(tags=['train1'])
    #test_X, test_Y, _ = get_db_data(tags=['train2'])
    train_X, train_Y, _ = get_db_data(tags=['train1', 'train2'])
    test_X, test_Y, _ = get_db_data(tags=['train3'])

    model = BigModel(columns, n_est=500)
    R2 = model.train_test(train_X, train_Y, test_X)
    print get_rmse(test_Y, R2)