keras 抽取模型中的一部分并把数据存入txt
程序员文章站
2022-03-26 15:53:05
...
for i in range(len(X)):
mcc.RNAEs[i, 0].load_weights('save/3-sourcesRNAES/ae_weights%d.h5' % (i + len(mcc.RNAEs) * 2)) #加载训练好的参数
sub_model = Model(inputs=mcc.RNAEs[i, 0].inputs, outputs=mcc.RNAEs[i, 0].layers[4].output) #取出到layers[4]的运算层
features = sub_model.predict(X[i]) #取出到layers[4]的运算结果
np.savetxt('./features/r5/%d_nae.txt' %(1*i) , features)
np.savetxt('./features/r5/y.txt',y)
print("***************save************************")