fix: restore save call

This commit is contained in:
hexastack 2024-11-28 20:51:36 +01:00
parent 896c546000
commit 3f5b98cc72

View File

@ -125,6 +125,7 @@ class SlotFiller(tfbp.Model):
# Persist the model # Persist the model
self.extra_params["slot_names"] = slot_names self.extra_params["slot_names"] = slot_names
self.extra_params["synonym_map"] = self.data_loader.get_synonym_map() self.extra_params["synonym_map"] = self.data_loader.get_synonym_map()
self.save()
@tfbp.runnable @tfbp.runnable
def evaluate(self): def evaluate(self):