main.py

import numpy as np
from opts import options


def random_dataset():
    """
    initializes a training and
    a testing dataset in the form
    of numpy arrays
    """
    np.random.seed(8675309)
    return (
        np.random.randn(10000, 10),
        np.random.randint(5, size=(10000, 1)),
    )


def main():
    method = input("\nChoose a method to test: ").lower()
    try:
        func = options[method]
    except KeyError:
        raise f"Invalid method \"{method}\". Try one of these\n{list(options.keys())}"

    X, y = random_dataset()
    result = func(X, y)


if __name__ == "__main__":
    main()