def save(self):
"""
Save all parameters to file
:param filename: file to save to
"""
d = {"type": self.model_type, "labels": self.labels, "model": dict(self.model), "is_frozen": self.is_frozen}
if not self.is_frozen:
d.update({"_min_update": self._min_update, "_update_index": self._update_index})
save_dict(self.filename, d)
def save(self):
"""
Save all parameters to file
:param filename: file to save to
"""
d = {
"type": "dense",
"labels": self.labels,
"model": self.model,
"is_frozen": self.is_frozen,
}
if not self.is_frozen:
d.update({
"_update_index": self._update_index,
})
save_dict(self.filename, d)
def save(self):
"""
Save all parameters to file
:param filename: file to save to
"""
d = {
"type": self.model_type,
"labels": self.labels,
"is_frozen": self.is_frozen,
}
save_dict(self.filename, d)
self.init_model()
with open(self.filename + ".json", "w") as f:
f.write(self.model.to_json())
try:
self.model.save_weights(self.filename + ".h5", overwrite=True)
except ValueError as e:
print("Failed saving model weights: %s" % e)