def main2(max_iter):
    # test Momentum optimizer

    n_classes, X_train, y_train = get_data()