-
Notifications
You must be signed in to change notification settings - Fork 0
/
madmenu.py
76 lines (67 loc) · 1.74 KB
/
madmenu.py
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
"""
Menu for Madline Neural Network
"""
import os
import sys
import time
import madaline as m
"""
Function to create the interactive menu for the user
"""
def menu():
#os.system('clear')
print("\nPlease choose from the options below:")
print("1. Train using a training data file")
print("2. Test/Deploy Neural Net")
print("3. Exit\n")
return
"""
Function that calls the selected method to run.
Arg:
choice - integer value for menu option
"""
def pick_one(choice):
try:
if choice in menu_choice:
vals = menu_choice[choice]()
# User entered number that's not valid
except ValueError:
print("Error: Invalid selection")
time.sleep(1)
menu()
except KeyboardInterrupt:
os.system('clear')
menu_choice[3]()
# print(vals)
return vals
"""
Print initial greeting and get input file
"""
def greetIn():
print('Welcome to my Madaline Neural Net!\n')
try:
# store as a input file
infile = input('Enter the data input file name: ')
if os.path.isfile(infile) and infile.endswith('.txt'): # return True if it is a file
return infile
except KeyboardInterrupt:
os.system('clear')
menu_choice[3]()
'''
except Error: # file not real or not a .txt file
print('The file you entered does not exist.')
print('Please enter a txt file')
time.sleep(1)
os.system('clear')
greetIn()
'''
"""
After we've trained the NN, we'll find out if we're going to test or
"""
# Options for menu
menu_choice = {"menu": menu,
1: m.initializeIt, # Call main from madaline.py
2: m.testIt,
#3: file_test,
3: sys.exit
}