文章目录
遇到问题
今天在刷 deep learning with python second edition 的时候,在NLP章节遇到了自己没理解的知识:keras中map函数的用法
text_only_train_ds = train_ds.map(lambda x, y: x)
解决
这里map的作用是:让CPU可以异步计算。在GPU处理上一个batch的数据时,CPU已经在处理下一个batch的预处理了。这样GPU处理完这个batch之后,可以无需等待即可开始下一个batch的计算。
那么后面lambda表达式labmda x, y: x
的作用是什么呢?lambda表达式会返回冒号后面的表达式,这里是x,也就是说返回x,丢弃了y(这里是标签)。
来自StackOverflow的问题
答案
参考
Map function in Keras