欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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