main.py

import numpy as np


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: str, X: np.array, y: np.array):
    pass



if __name__ == "__main__":
    method = input("\nChoose a method to test: ").lower()

    X, y = random_dataset()
    main(
        method=method, 
        X=X, 
        y=y,
    )