experiments/momentum_regimes.py [536:643]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            max_iter=max_iter,
            kernel="RBF",
            kernel_sigma=1.0,
            kernel_nu=0.5,
            random_state=random_state,
            verbose=0,
        )

    return model


def filter_run_names_exp(run_names, solver, sketch_size_formula):
    """Filter runs for given solver and sketch_size_formula"""
    solver_run_names = [
        run_name for run_name in run_names if solver == run_name.split(" | ")[1]
    ]
    run_names = [n for n in solver_run_names if sketch_size_formula in n]

    # To alwayshave in the same order:
    # no momentum, then constant momentum, then increasing momentum
    run_names.sort(reverse=True)
    run_names = [run_names[0], run_names[2], run_names[1]]
    return run_names


def format_run_names_exp(run_names, solver, sketch_size_formula):
    """
    Selects and formats run names based on run name, solver and sketch size
    """
    # formatted_run_names = ["|".join(r.split("|")[:-1]).strip() for r in run_names]
    # formatted_run_names = [r.split(" |")[0] for r in run_names]
    formatted_run_names = []
    for r in run_names:
        mom_version = r.split(" |")[0]
        if mom_version == "inc_mom":
            formatted_run_names.append("increasing momentum")
        elif mom_version == "cst_mom":
            formatted_run_names.append("constant momentum")
        elif mom_version == "no_mom":
            formatted_run_names.append("no momentum")
        else:
            raise ValueError("Unknown momentum version")
    return formatted_run_names


def filter_sketch_sizes(sketch_sizes, sketch_size_formulas):
    """Filter duplicate sketch sizes and merge the formulas"""
    filtered_sketch_sizes = sketch_sizes.copy()
    filtered_sketch_size_formulas = sketch_size_formulas.copy()
    if len(sketch_sizes) not in [1, 2, 3]:
        raise ValueError("Works only with 2 or 3 sketch sizes")
    elif len(sketch_sizes) == 2:
        if len(np.unique(filtered_sketch_sizes)) < len(filtered_sketch_sizes):
            if filtered_sketch_sizes[0] == filtered_sketch_sizes[1]:
                filtered_sketch_size_formulas[1] = (
                    filtered_sketch_size_formulas[0]
                    + " = "
                    + filtered_sketch_size_formulas[1]
                )
                del filtered_sketch_sizes[0]
                del filtered_sketch_size_formulas[0]
    elif len(sketch_sizes) == 3:
        if len(np.unique(filtered_sketch_sizes)) < len(filtered_sketch_sizes):
            idx_to_del = []
            if (
                filtered_sketch_sizes[0] == filtered_sketch_sizes[1]
                and filtered_sketch_sizes[1] == filtered_sketch_sizes[2]
            ):
                idx_to_del = [0, 1]
                filtered_sketch_size_formulas[2] = (
                    filtered_sketch_size_formulas[0]
                    + " = "
                    + filtered_sketch_size_formulas[1]
                    + " = "
                    + filtered_sketch_size_formulas[2]
                )
            elif filtered_sketch_sizes[0] == filtered_sketch_sizes[1]:
                idx_to_del = [0]
                filtered_sketch_size_formulas[1] = (
                    filtered_sketch_size_formulas[0]
                    + " = "
                    + filtered_sketch_size_formulas[1]
                )
            elif filtered_sketch_sizes[1] == filtered_sketch_sizes[2]:
                idx_to_del = [1]
                filtered_sketch_size_formulas[2] = (
                    filtered_sketch_size_formulas[1]
                    + " = "
                    + filtered_sketch_size_formulas[2]
                )
            elif filtered_sketch_sizes[0] == filtered_sketch_sizes[2]:
                idx_to_del = [0]
                filtered_sketch_size_formulas[2] = (
                    filtered_sketch_size_formulas[0]
                    + " = "
                    + filtered_sketch_size_formulas[2]
                )

            for index in sorted(idx_to_del, reverse=True):
                del filtered_sketch_sizes[index]
                del filtered_sketch_size_formulas[index]

    return filtered_sketch_sizes, filtered_sketch_size_formulas


@click.command()
@click.argument("dataset_name")
@click.option(
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



experiments/regularization_on_momentum.py [492:598]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            max_iter=max_iter,
            kernel="RBF",
            kernel_sigma=1.0,
            kernel_nu=0.5,
            random_state=random_state,
            verbose=0,
        )

    return model


def filter_run_names_exp(run_names, solver, sketch_size_formula):
    """Filter runs for given solver and sketch_size_formula"""
    solver_run_names = [
        run_name for run_name in run_names if solver == run_name.split(" | ")[1]
    ]
    run_names = [n for n in solver_run_names if sketch_size_formula in n]

    # To alwayshave in the same order:
    # no momentum, then constant momentum, then increasing momentum
    run_names.sort(reverse=True)
    run_names = [run_names[0], run_names[2], run_names[1]]
    return run_names


def format_run_names_exp(run_names, solver, sketch_size_formula):
    """
    Selects and formats run names based on run name, solver and sketch size
    """
    # formatted_run_names = ["|".join(r.split("|")[:-1]).strip() for r in run_names]
    formatted_run_names = []
    for r in run_names:
        mom_version = r.split(" |")[0]
        if mom_version == "inc_mom":
            formatted_run_names.append("increasing momentum")
        elif mom_version == "cst_mom":
            formatted_run_names.append("constant momentum")
        elif mom_version == "no_mom":
            formatted_run_names.append("no momentum")
        else:
            raise ValueError("Unknown momentum version")
    return formatted_run_names


def filter_sketch_sizes(sketch_sizes, sketch_size_formulas):
    """Filter duplicate sketch sizes and merge the formulas"""
    filtered_sketch_sizes = sketch_sizes.copy()
    filtered_sketch_size_formulas = sketch_size_formulas.copy()
    if len(sketch_sizes) not in [1, 2, 3]:
        raise ValueError("Works only with 2 or 3 sketch sizes")
    elif len(sketch_sizes) == 2:
        if len(np.unique(filtered_sketch_sizes)) < len(filtered_sketch_sizes):
            if filtered_sketch_sizes[0] == filtered_sketch_sizes[1]:
                filtered_sketch_size_formulas[1] = (
                    filtered_sketch_size_formulas[0]
                    + " = "
                    + filtered_sketch_size_formulas[1]
                )
                del filtered_sketch_sizes[0]
                del filtered_sketch_size_formulas[0]
    elif len(sketch_sizes) == 3:
        if len(np.unique(filtered_sketch_sizes)) < len(filtered_sketch_sizes):
            idx_to_del = []
            if (
                filtered_sketch_sizes[0] == filtered_sketch_sizes[1]
                and filtered_sketch_sizes[1] == filtered_sketch_sizes[2]
            ):
                idx_to_del = [0, 1]
                filtered_sketch_size_formulas[2] = (
                    filtered_sketch_size_formulas[0]
                    + " = "
                    + filtered_sketch_size_formulas[1]
                    + " = "
                    + filtered_sketch_size_formulas[2]
                )
            elif filtered_sketch_sizes[0] == filtered_sketch_sizes[1]:
                idx_to_del = [0]
                filtered_sketch_size_formulas[1] = (
                    filtered_sketch_size_formulas[0]
                    + " = "
                    + filtered_sketch_size_formulas[1]
                )
            elif filtered_sketch_sizes[1] == filtered_sketch_sizes[2]:
                idx_to_del = [1]
                filtered_sketch_size_formulas[2] = (
                    filtered_sketch_size_formulas[1]
                    + " = "
                    + filtered_sketch_size_formulas[2]
                )
            elif filtered_sketch_sizes[0] == filtered_sketch_sizes[2]:
                idx_to_del = [0]
                filtered_sketch_size_formulas[2] = (
                    filtered_sketch_size_formulas[0]
                    + " = "
                    + filtered_sketch_size_formulas[2]
                )

            for index in sorted(idx_to_del, reverse=True):
                del filtered_sketch_sizes[index]
                del filtered_sketch_size_formulas[index]

    return filtered_sketch_sizes, filtered_sketch_size_formulas


@click.command()
@click.argument("dataset_name")
@click.option(
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -