TensorFlow系列——一些api的使用场景及方式

本文介绍了在TensorFlow中如何利用string_split处理tfrecord文件中的多类别特征,结合tf.map_fn实现高效操作。同时,讨论了如何使用filter对数据集进行筛选,并展示了keras模型summary()的功能。此外,还详细阐述了如何在模型中添加正则化器,如l2正则化,以及其在训练过程中的应用。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1、string_split:

在制作tfrecord的时候,将多个类别特征使用分隔符组装到一起,用一个特征列来存储,降低存储代价。而在训练时候使用string_split将其分隔,分别处理。

参考:

https://www.tensorflow.org/api_docs/python/tf/compat/v1/string_split

结合使用的api:

tf.map_fn

2、filter

对TensorFlow的dataset进行过滤

参考:

https://stackoverflow.com/questions/48825785/how-can-i-filter-tf-data-dataset-by-specific-values

3、keras里的summary()

打印出模型的结构

https://stackoverflow.com/questions/61511786/question-regarding-how-to-directly-apply-softmax-onto-a-logits-with-tf-keras-act

4、为模型添加正则化器:

regularizer = tf.keras.regularizers.l2(l2_factor)

使用时:mylayer.add_loss(self.regularizer(column_weight)),其中mylayer是自定义的tf.keras.layers.Layer,而column_weight为参数tensor。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值