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