mlp.py

"""
Main script of the project.
"""

import numpy as np

from mlp.parser import create_parser
from mlp import split, train, predict


def main() -> None:
    """
    Main function of the script.
    """

    parser = create_parser()
    args = parser.parse_args()

    if args.seed:
        np.random.seed(args.seed)

    match args.command:
        case "split":
            split(args.dataset, args.test_size, args.out_dir)
        case "train":
            train(
                args.dataset,
                args.model,
                args.out_dir,
                args.no_plot,
                args.plot_n,
                args.plot_raw,
            )
        case "predict":
            predict(args.dataset, args.model)


if __name__ == "__main__":
    main()