- 使用Keras框架时,自定义函数
def cos_value(x):
return tf.cos(x) + 0.5 # 必须 tensor类型
model = Sequential([
layers.GRU(units=512, input_shape=(timesteps, 1), return_sequences=True,
kernel_regularizer=tf.keras.regularizers.l2(0.001)),
layers.GRU(units=512, activation=cos_value, kernel_regularizer=tf.keras.regularizers.l2(0.001)),
layers.Dense(units=5)
])
此时只能通过张量的形式保存模型