首先上代码:
@tf.keras.utils.register_keras_serializable('my_package')
def mysplit(inputs):
return tf.strings.split(inputs)
input3 = tf.keras.layers.Input(shape=(1,),name="string_test",dtype=tf.string)
my_string_lookup = tf.keras.layers.TextVectorization(vocabulary=["世界","你","good", "d"],split=mysplit)(input3)
model = tf.keras.Model(inputs=[input3], outputs=my_string_lookup)
model.save("mytest.h5")
my_model = tf.keras.models.load_model("mytest.h5")
print(my_model.predict(["世界 你 good ya"]))
TextVectorization是相对于StringLookup更为复杂的文本处理层,除了支持StringLookup对词库的各种编码方式(indices、one-hot、tf-idf、count)外,还支持对构造词库和请求内容时的标准化、自动分词、去除标点服务、去重多余空格等处理。
其中标准化、自动分词都可以通过自定义方式进行处理,但是自定义的处理算子需要进行序列化装饰器进行装饰:t

文章介绍了如何在TensorFlow中使用@tf.keras.utils.register_keras_serializable装饰自定义函数,如mysplit,以配合TextVectorization层进行文本处理,包括分词、标准化和自定义分隔符。TextVectorization层不仅支持StringLookup的功能,还提供额外的文本预处理选项。示例展示了模型的保存和加载,并预告了接下来将探讨如何利用indices方法进行embedding查找。
最低0.47元/天 解锁文章
4547

被折叠的 条评论
为什么被折叠?



