1. Home
  2. Docs
  3. Python
  4. Tensorflow
  5. 權重-讀取模型

權重-讀取模型


try:
    with open('model2.json', 'r') as json_file:
        loaded_model_json = json_file.read()
        json_file.close()
        model = model_from_json(loaded_model_json)
        # 讀取模型權重
        model.load_weights("model2.h5")
        learning_rate = 0.0001
        opt1 = tf.keras.optimizers.Nadam(lr=learning_rate)
        model.compile(loss='mse', optimizer=opt1, metrics=['mae'])
except IOError:
        print("error")