diff --git a/src/setup/interactor.py b/src/setup/interactor.py index a7e9e41..0908888 100644 --- a/src/setup/interactor.py +++ b/src/setup/interactor.py @@ -75,6 +75,9 @@ def measures_input() -> list: # Appends every Complexity Measure in measures list return [cm(measure) for measure in input_Stream] +def generation_input() -> int: + return int(input("Com quantas gerações você deseja otimizar o dataset?\n")) + def __input_with_default__(input_text: str, default_value, data_type: type): try: return data_type(input(input_text)) diff --git a/src/setup/setup_framework.py b/src/setup/setup_framework.py index bf17fff..802840b 100644 --- a/src/setup/setup_framework.py +++ b/src/setup/setup_framework.py @@ -45,6 +45,7 @@ def setup_interative() -> dict: # TODO: Extract Measures from the real dataset options['filename'] = interactor.filename_input() + options['NGEN'] = interactor.generation_input() # Separating Measures if measures != None: