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
