machine_learning_projects/main.py at master · Andrei-Chiorian/machine_learning_projects · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
from machine_learning_algorithms.linear_regression import linear_regression
from utils.formatting.colors import Colors
from machine_learning_algorithms.logistic_regression import logistic_regression
from data_visualization.data_visualization import data_visualization
if __name__ == '__main__':

    while True:
        print(Colors.HEADER, Colors.UNDERLINE, "\nMACHINE LEARNING ALGORITHM'S \n\n", Colors.ENDC)

        print(Colors.OKBLUE, "- 1 - Linear Regression", Colors.ENDC)

        print(Colors.OKBLUE, "\n - 2 - Logistic Regression", Colors.ENDC)

        print(Colors.OKBLUE, "\n - 3 - Dataset visualization", Colors.ENDC)

        print(Colors.OKBLUE, "\n - 4 - Linear Regression", Colors.ENDC)

        algorithm = input(Colors.OKCYAN + "\n¿Enter the number of the machine learning algorithm or exercise would you like to run?"
                                          "\n" + Colors.ENDC)

        if algorithm:
            match int(algorithm):
                case 1:
                    linear_regression()
                    print(Colors.OKCYAN, "\n\nLinear Regression Algorithm example completed", Colors.ENDC)
                    print(Colors.FAIL, "\n\n-------------------------------------------------------------------------"
                                       "--------------", Colors.ENDC)
                case 2:
                    logistic_regression()
                    print(Colors.OKCYAN, "\n\nLogistic Regression Algorithm example completed", Colors.ENDC)
                    print(Colors.FAIL, "\n\n-------------------------------------------------------------------------"
                                       "--------------", Colors.ENDC)
                case 3:
                    data_visualization()
                    print(Colors.OKCYAN, "\n\nDataset visualization example completed", Colors.ENDC)
                    print(Colors.FAIL, "\n\n-------------------------------------------------------------------------"
                                       "--------------", Colors.ENDC)
                case _:
                    print(Colors.OKCYAN, "Opción no válida", Colors.ENDC)