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")