Theano中怎么保存和加載模型

小億
82
2024-03-25 15:09:43

在Theano中保存和加載模型，通常使用numpy庫(kù)中的np.save和np.load函數(shù)。以下是保存和加載模型的示例代碼：

保存模型：

import numpy as np
import theano
import theano.tensor as T

# 定義模型
x = T.matrix('x')
y = T.matrix('y')
z = x + y

# 編譯函數(shù)
add = theano.function([x, y], z)

# 保存模型
np.save('model.npy', add)

加載模型：

import numpy as np
import theano

# 加載模型
add = np.load('model.npy').item()

# 使用模型
result = add([[1, 2], [3, 4]], [[5, 6], [7, 8]])
print(result)

上述代碼中，我們定義了一個(gè)簡(jiǎn)單的模型，并使用np.save函數(shù)保存模型。然后通過np.load函數(shù)加載模型，并使用加載的模型進(jìn)行預(yù)測(cè)。在實(shí)際應(yīng)用中，可以根據(jù)具體的模型和數(shù)據(jù)進(jìn)行相應(yīng)的保存和加載操作。

0 踩