def load_model_by_id(self, model_id): """Get the model by model_id Parameters ---------- model_id : int model index Returns ------- load_model : Graph the model graph representation """ with open(os.path.join(self.path, str(model_id) + ".json")) as fin: json_str = fin.read().replace("\n", "") load_model = json_to_graph(json_str) return load_model
def zmq_server_func(self): while True: try: message = self.socket.recv_pyobj() if message["type"] == "get_next_parameter": ret = {} lock.acquire() ret["tuner"] = self.tuner self.socket.send_pyobj(ret) lock.release() elif message["type"] == "generated_parameter": self.current_jobs += 1 print("New model generated, current jobs = " + str(self.current_jobs)) if not "parameters" in message: self.socket.send_pyobj("nothing") continue parameter_id = message["parameter_id"] father_id = message["father_id"] json_params = message["parameters"] lock.acquire() x, y, model_id = self.tuner.total_data[parameter_id] generated_graph = json_to_graph(json_params) self.tuner.set_descriptors(model_id, generated_graph) self.tuner.total_data[parameter_id] = (json_params, father_id, model_id) _trial_params[parameter_id] = json_params lock.release() self.socket.send_pyobj("nothing") elif message["type"] == "FINAL": self.socket.send_pyobj("nothing") self._handle_final_metric_data(message) except Exception as e: print('error:', e) sys.exit()
def build_graph_from_json(ir_model_json): """build model from json representation """ graph = json_to_graph(ir_model_json) model = graph.produce_keras_model() return model