我指的是一些链接,以删除模型的最后一层并保存以供将来预测。
How to remove the last layer from trained model in Tensorflow
只是想知道是否还有更多方法可以做到这一点。例如
from tensorflow.keras import layers, losses
from tensorflow.keras.models import Model
model = tf.keras.Sequential([
layers.Dense(64,activation='relu',input_shape=(None,784)),
layers.Dense(128, activation='relu'),
layers.Dense(10,activation='sigmoid')
])
model.compile(optimizer='adam', loss=losses.SparseCategoricalCrossentropy())
model1 = Model(inputs = model.input, outputs= model.layers[-2].output)
model1.save()
model1
是新模型。
我的假设是新模型将提供经过训练的权重。如果我错了,请纠正我。