def load(self):
"""
Load all parameters from file
:param filename: file to load from
"""
d = load_dict(self.filename)
model_type = d.get("type")
assert model_type == "dense", "Model type does not match: %s" % model_type
self.labels = list(d["labels"])
self.model = d["model"]
self.is_frozen = d["is_frozen"]
if not self.is_frozen:
self._update_index = d["_update_index"]
def load(self):
"""
Load all parameters from file
:param filename: file to load from
"""
d = load_dict(self.filename)
model_type = d.get("type")
assert model_type == self.model_type, "Model type does not match: %s" % model_type
self.labels = list(d["labels"])
self.is_frozen = d["is_frozen"]
with open(self.filename + ".json") as f:
self.model = model_from_json(f.read())
try:
self.model.load_weights(self.filename + ".h5")
except KeyError as e:
print("Failed loading model weights: %s" % e)
self.compile()
