Tensorflow - 네트워크와 웨이트를 따로 저장하고 불러오는 방법

네트워크만 저장하고 불러오기

 

 

 

 

model.to_json()

 

# 저장하기

my_network = model.to_json()

with open('my_network.json', 'w') as json_file :
  json_file.write(my_network)

 

# 불러오기

with open('my_network.json', 'r') as json_file :
  my_net2 = json_file.read()
  
my_net2

 

 

 

 

웨이트만 저장하고 불러오기

 

 

 

 

# 저장하기

model.save_weights('fashion_mnist_weight.h5')

 

 

# 불러오기

model4.load_weights('fashion_mnist_weight.h5')

 

model4.predict(X_test)