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()