experiments/different_sketch_sizes_time_bar_plots.py [576:618]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            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
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



experiments/regularization_on_momentum.py [492:533]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            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
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -