def train_model()

in quickstarts/microsoft.machinelearningservices/machine-learning-job-create-sweep-job/scripts/main.py [0:0]

• 11 lines of code
• 1 McCabe index (conditional complexity)

def train_model(params, num_boost_round, X_train, X_test, y_train, y_test):
    # create lightgbm datasets
    train_data = lgbm.Dataset(X_train, label=y_train)
    test_data = lgbm.Dataset(X_test, label=y_test)

    # train model
    model = lgbm.train(
        params,
        train_data,
        num_boost_round=num_boost_round,
        valid_sets=[test_data],
        valid_names=["test"],
    )

    # return model
    return model