Ejemplo n.º 1
0
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')
Ejemplo n.º 2
0
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')
Ejemplo n.º 3
0
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')
Ejemplo n.º 4
0
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')