结果如图10-61所示。
下面看看能否优化该模型以改进其效果。我们可以更改超参数或使用其他优化工具，比如Adam，也可以向informative terms中添加其他术语。此数据集有一个完整的词汇表文件，其中包含30716个术语，该文件为https://storage.googleapis.com/mledu-datasets/sparse-data-embedding/terms.txt，如图10-62所示。我们可以从该词汇表文件中挑选出其他术语，也可以通过categorical\_column\_with\_vocabulary\_file特征列使用整个词汇表文件。
训练代码如下：
第10章 神经网络
369