예제 #1
0
def train_model(training_file_name):
    data = common.read_file(training_file_name)
    x_train = data[:, features]
    y_train = data[:, 1]
    xgb = XGBClassifier()
    xgb.classes_ = [0, 1]
    xgb = xgb.fit(x_train, y_train)
    scores = cross_val_score(xgb, x_train, y_train, cv=10, scoring='accuracy')
    print("Scores gotten using AdaBoost classifier (# of estimators=" +
          str(25) + ")")
    print(scores)
    print(np.mean(scores))
    return xgb