快捷導(dǎo)航

在keras下實現(xiàn)多個模型的融合方式

更新時間：2020年05月23日 15:58:59 作者：小風(fēng)風(fēng)12580

這篇文章主要介紹了在keras下實現(xiàn)多個模型的融合方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

在網(wǎng)上搜過發(fā)現(xiàn)關(guān)于keras下的模型融合框架其實很簡單，奈何網(wǎng)上說了一大堆，這個東西官方文檔上就有，自己寫了個demo：

# Function：基于keras框架下實現(xiàn)，多個獨立任務(wù)分類
# Writer: PQF
# Time: 2019/9/29

import numpy as np
from keras.layers import Input, Dense
from keras.models import Model
import tensorflow as tf

# 生成訓(xùn)練集
dataset_size = 128*3
rdm = np.random.RandomState(1)
X = rdm.rand(dataset_size,2)
Y1 = [[int(x1+x2<1)] for (x1,x2) in X]
Y2 = [[int(x1+x2*x2<0.5)] for (x1,x2) in X]

X_train = X[:-2]
Y_train1 = Y1[:-2]
Y_train2 = Y2[:-2]

X_test = X[-2:dataset_size]
Y_test1 = Y1[-2:dataset_size]
Y_test2 = Y2[-2:dataset_size]

#網(wǎng)絡(luò)一
input = Input(shape=(2,))
x = Dense(units=16,activation='relu')(input)
output = Dense(units=1,activation='sigmoid',name='output1')(x)

#網(wǎng)絡(luò)二
input2 = Input(shape=(2,))
x2 = Dense(units=16,activation='relu')(input2)
output2 = Dense(units=1,activation='sigmoid',name='output2')(x2)

#模型合并
model = Model(inputs=[input,input2],outputs=[output,output2])
model.summary()

model.compile(optimizer='rmsprop',loss='binary_crossentropy',loss_weights=[1.0,1.0])
model.fit([X_train,X_train],[Y_train1,Y_train2],batch_size=48,epochs=200)

print('x_test is :\n')
print(X_test)
print('y_test1 is :\n')
print(Y_test1)
print('y_test2 is :\n')
print(Y_test2)

predict = model.predict([X_test,X_test])
print('prediction is : \n')
print(predict[0])
print(predict[1])

補充知識：keras的融合層使用理解

最近開始研究U-net網(wǎng)絡(luò)，其中接觸到了融合層的概念，做個筆記。