Beispiel #1
0
def train_and_evaluate_xgboost_model(plot_graph=False):
    data_dict = load_and_split()
    train_set = data_dict.get('train_set').to_numpy()
    test_set = data_dict.get('test_set').to_numpy()
    reg = cross_val_xgboost_model(train_set)
    test_model(reg, test_set, plot_graph)
Beispiel #2
0
def validate_xgboost_model():
    data_dict = load_and_split()
    train_set = data_dict.get('train_set').to_numpy()
    reg = cross_val_xgboost_model(train_set)