Spaces:
Running on Zero
Running on Zero
| from misc.enums import Mode | |
| from misc.Cache import setup_chache | |
| from machine_learning.predict import run_prediction | |
| from machine_learning.train import run_optimization, run_training | |
| from setup.configProcessor import validate_config, select_mode_from_config, use_cache, get_model_parameter_dict | |
| from setup.generalSetup import seed_all | |
| from output.outputFileWriting import write_predictions_to_file | |
| from src.data_processing.post_processing import get_averaged_predictions | |
| if __name__ == "__main__": | |
| print('I am running') | |
| seed_all(seed=42) | |
| validate_config() | |
| mode = select_mode_from_config() | |
| print(f'Selected mode is {mode.name}') | |
| if use_cache(): | |
| setup_chache() | |
| match mode: | |
| case Mode.PREDICT: | |
| model_parameters = get_model_parameter_dict(only_first_value=True) | |
| predictions = run_prediction(model_parameters) | |
| avg_predictions = get_averaged_predictions(predictions) | |
| write_predictions_to_file(avg_predictions) | |
| case Mode.OPTIMIZE: | |
| model_parameters = get_model_parameter_dict(only_first_value=False) | |
| run_optimization(model_parameters) | |
| case Mode.TRAIN: | |
| model_parameters = get_model_parameter_dict(only_first_value=True) | |
| run_training(model_parameters) | |