def main(main_path):
"""
Parse command line arguments and run either train or test mode
"""
# Output program header
with open('mimosas.txt', 'r') as f:
contents = f.read()
print(contents)
print()
for line in header:
print(line)
print('\n')
# Parse config file
parameters = Parameters(main_path)
# Run indicated models
if ('DECISION_TREE' in parameters.config.sections()):
run_decision_tree(parameters)
if ('RANDOM_FOREST' in parameters.config.sections()):
run_random_forest(parameters)
if ('FEED_FORWARD' in parameters.config.sections()):
run_feed_forward_nn(parameters)
def __init__(self):
self.params = Parameters()
self.trials = 0
self.rounds = 3
