| 92 | |
| 93 | |
| 94 | def init_error( |
| 95 | modelFolder, |
| 96 | dataFolder, |
| 97 | params, |
| 98 | nn=None, |
| 99 | data_dict=None, |
| 100 | var_dict=None, |
| 101 | params_dict=None, |
| 102 | params_list=None, |
| 103 | ): |
| 104 | print("INFO: Using modelFolder : ", modelFolder) |
| 105 | print("INFO: Using dataFolder : ", dataFolder) |
| 106 | print("INFO: Loading from config file") |
| 107 | |
| 108 | if not nn: |
| 109 | with open(os.path.join(modelFolder, "config.json")) as json_file: |
| 110 | configDict = json.load(json_file) |
| 111 | nn = initialize_nn_from_params_config(params, configDict) |
| 112 | if not data_dict: |
| 113 | data_dict = make_data_dict(dataFolder, params_list) |
| 114 | if not var_dict or not params_dict: |
| 115 | var_dict, params_dict = make_var_params_from_data(nn, data_dict) |
| 116 | nn = safe_load(nn, os.path.join(modelFolder, "best.weights.h5")) |
| 117 | |
| 118 | return nn, data_dict, var_dict, params_dict |
| 119 | |
| 120 | |
| 121 | if __name__ == "__main__": |