Herowaree

Load and save model

Nov 25th, 2021 (edited)
183
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.00 KB | None | 0 0
  1. from keras.models import model_from_json
  2.  
  3. import os
  4. import json
  5.  
  6. def load_model(dir_path):
  7. # Abrir o arquivo
  8. json_file = open(dir_path + "/model.json", "r")
  9.  
  10. # Carregar o arquivo
  11. loaded_model_json = json_file.read()
  12.  
  13. # Fechar o arquivo
  14. json_file.close()
  15.  
  16. # Carregar o modelo
  17. loaded_model = model_from_json(loaded_model_json)
  18.  
  19. # Carregar os pesos do modelo
  20. loaded_model.load_weights(dir_path + "/model.h5")
  21.  
  22. return loaded_model
  23.  
  24. def save_model(dir_path, model):
  25. # Verificar se já existe o arquivo do modelo (se existir, só precisa atualizar os pesos)
  26. if os.path.isfile(dir_path + "/model.json"):
  27. # Deletar o arquivo de pesos antigo
  28. os.remove(dir_path + "/model.h5")
  29. else:
  30. # Converter o modelo em um arquivo .json
  31. model_json = model.to_json()
  32.  
  33. # Salvar o arquivo .json
  34. with open(dir_path + "/model.json", "w") as json_file:
  35. json_file.write(model_json)
  36.  
  37. # Salvar os pesos
  38. model.save_weights(dir_path + "/model.h5")
Add Comment
Please, Sign In to add comment