Python run_lgbの例

コード例 #1

def main():
    args = parse_args()
    feature_matrix = joblib.load(
        '../data/02_featuretools/feature_matrix.joblib')
    app_train = feature_matrix.dropna(subset=['TARGET'])
    app_test = feature_matrix[feature_matrix['TARGET'].isnull()].drop('TARGET',
                                                                      axis=1)
    run_lgb(args, app_train, app_test, '12_featuretools')

コード例 #2

def main():
    args = parse_args()
    app_train = joblib.load('../data/01_labelencoding/application_train.joblib')
    app_test = joblib.load('../data/01_labelencoding/application_test.joblib')
    features = read_all('../data/21_dimlstm')
    for feature in features.values():
        app_train = app_train.merge(feature)
        app_test = app_test.merge(feature)
    
    run_lgb(args, app_train, app_test, '31_dimlstm')

コード例 #3

ファイル: 32_ft_dimlstm.py プロジェクト: Sandy4321/HomeCredit

def main():
    args = parse_args()
    feature_matrix = joblib.load(
        '../data/02_featuretools/feature_matrix.joblib')
    app_train = feature_matrix.dropna(subset=['TARGET'])
    app_test = feature_matrix[feature_matrix['TARGET'].isnull()].drop('TARGET',
                                                                      axis=1)
    features = read_all('../data/21_dimlstm')
    for feature in features.values():
        app_train = app_train.merge(feature)
        app_test = app_test.merge(feature)

    run_lgb(args, app_train, app_test, '32_ft_dimlstm')

コード例 #4

def main():
    args = parse_args()
    app_train = joblib.load('../data/01_labelencoding/application_train.joblib')
    app_test = joblib.load('../data/01_labelencoding/application_test.joblib')
    run_lgb(args, app_train, app_test, '11_baseline')