Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from keras.models import model_from_json
- import os
- import json
- def load_model(dir_path):
- # Abrir o arquivo
- json_file = open(dir_path + "/model.json", "r")
- # Carregar o arquivo
- loaded_model_json = json_file.read()
- # Fechar o arquivo
- json_file.close()
- # Carregar o modelo
- loaded_model = model_from_json(loaded_model_json)
- # Carregar os pesos do modelo
- loaded_model.load_weights(dir_path + "/model.h5")
- return loaded_model
- def save_model(dir_path, model):
- # Verificar se já existe o arquivo do modelo (se existir, só precisa atualizar os pesos)
- if os.path.isfile(dir_path + "/model.json"):
- # Deletar o arquivo de pesos antigo
- os.remove(dir_path + "/model.h5")
- else:
- # Converter o modelo em um arquivo .json
- model_json = model.to_json()
- # Salvar o arquivo .json
- with open(dir_path + "/model.json", "w") as json_file:
- json_file.write(model_json)
- # Salvar os pesos
- model.save_weights(dir_path + "/model.h5")
Add Comment
Please, Sign In to add comment