classifiers.py 文件源码

def XGBoost(X, y):
    print("Iniciando treinamento do XGBoost")
    start_time = time.time()

    X_train, X_test, y_train, y_test = cross_validation.train_test_split(X, y, test_size=0.1, random_state=1)
    clf = xgb.XGBClassifier(learning_rate=0.15, n_estimators=170, nthread=6, max_depth=8, seed=0, silent=True,
                            subsample=0.85, colsample_bytree=0.85)
    clf.fit(X, y)
    score = clf.score(X_test, y_test)
    print("XGBoost score: ", score, "(", (time.time()-start_time)/60.0, "minutos )")

    return clf