keras載入模型

2021-09-10 16:05:04 字數 1825 閱讀 5332

說明:該程式是乙個包含兩個隱藏層的神經網路。演示如何載入乙個儲存好的模型。

資料集:mnist

from __future__ import print_function

#python提供了__future__模組,把下乙個新版本的特性匯入到當前版本,於是我們就可以在當前版本中測試一些新版本的特性。

import numpy as np

np.random.seed(1337) # for reproducibility

from keras.datasets import mnist

from keras.models import sequential

from keras.layers.core import dense,dropout,activation

from keras.optimizers import sgd,adam,rmsprop

from keras.utils import np_utils

需要載入load_model

from keras.models import load_model
變數初始化

batch_size = 128 

nb_classes = 10

nb_epoch = 20

準備資料

# the data, shuffled and split between train and test sets

(x_train, y_train), (x_test, y_test) = mnist.load_data()

x_train = x_train.reshape(60000, 784)

x_test = x_test.reshape(10000, 784)

x_train = x_train.astype('float32')

x_test = x_test.astype('float32')

x_train /= 255

x_test /= 255

print(x_train.shape[0], 'train samples')

print(x_test.shape[0], 'test samples')

#轉換類標號

# convert class vectors to binary class matrices

y_train = np_utils.to_categorical(y_train, nb_classes)

y_test = np_utils.to_categorical(y_test, nb_classes)

建立模型

#在現有的檔案中載入模型

model=load_model('mnist-mpl.h5')

#列印模型

model.summary()

訓練與評估

#編譯模型

model.compile(loss='categorical_crossentropy',

optimizer=rmsprop(),

metrics=['accuracy'])

模型評估

score = model.evaluate(x_test, y_test, verbose=0)

print('test score:', score[0])

print('test accuracy:', score[1])

TensorFlow中載入Keras模型

在tensorflow中,使用keras訓練好的模型或者keras自帶的預訓練模型 自定義模型 讀取模型或者載入預訓練模型,下面使用的是預訓練的vgg模型 model origin vgg16 weights imagenet 自定義某一層作為輸出 model new model inputs mo...

Keras模型載入之通過json檔案

匯入圖 匯入權重 示例import numpy as np from keras.models import model,load model from keras.models import model from json import json json file path to model.j...

keras評估模型

當建立好模型並且用來訓練之後,如何評估模型的好壞,準確度又如何呢?三種常用方法 1 使用自動驗證方法 在 fit 函式中增加乙個validation split引數,該引數用來進行驗證效果 該引數可以自由設定,一般設定為20 或者30 也就是測試集佔總資料集的20 或者30 的資料用來進行驗證,其餘...