使用Keras层的单词嵌入可视化按类型分组的电影

问题描述

我想在以下任务上寻求您的帮助。我训练了Keras顺序nn来预测电影的流派。电影可以属于一种或多种类型(例如['动作','喜剧'])。
对于神经网络,我使用了5个输入层和5个嵌入层。输入层之一具有属于数据集电影标题的单词(标记)。

我要完成的工作是根据流派绘制电影标题,以评估我的模型是否做出了不错的预测。由于一部电影可以属于多个类型,因此我可以过滤类型序列以使每个电影仅包含第一个类型(第一个类型是最主要的类型)。

我希望我的最终输出看起来像这样(取自article):

enter image description here

文章的作者根据类型对书名进行了标绘,我想改用电影标题

您可以找到我的Google colab笔记本here,以查看我已完成的工作以及嵌入层的权重。我的问题是,考虑到标题的每个单词都有词嵌入,我该如何按流派绘制电影标题

我也搜索Embedding Projector,但是我仍然不明白如何获取根据流派绘制电影标题所需的文件。您会在我的colab笔记本上注意到,我到达了一个虚拟数据帧,该数据帧具有电影标题列,该标题的嵌入以及每部电影的第一流派(而不是整个序列)。

虚拟数据帧:

enter image description here

我如何利用该数据集进行绘图,就像之前发布的图像一样?
预先感谢您的帮助和建议。

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)