问题描述
我想在以下任务上寻求您的帮助。我训练了Keras顺序nn来预测电影的流派。电影可以属于一种或多种类型(例如['动作','喜剧'])。
对于神经网络,我使用了5个输入层和5个嵌入层。输入层之一具有属于数据集电影标题的单词(标记)。
我要完成的工作是根据流派绘制电影标题,以评估我的模型是否做出了不错的预测。由于一部电影可以属于多个类型,因此我可以过滤类型序列以使每个电影仅包含第一个类型(第一个类型是最主要的类型)。
您可以找到我的Google colab笔记本here,以查看我已完成的工作以及嵌入层的权重。我的问题是,考虑到标题的每个单词都有词嵌入,我该如何按流派绘制电影标题。
我也搜索了Embedding Projector,但是我仍然不明白如何获取根据流派绘制电影标题所需的文件。您会在我的colab笔记本上注意到,我到达了一个虚拟数据帧,该数据帧具有电影标题列,该标题的嵌入以及每部电影的第一流派(而不是整个序列)。
虚拟数据帧:
我如何利用该数据集进行绘图,就像之前发布的图像一样?
预先感谢您的帮助和建议。
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)