文件名称:word源码java-Attention-Visualization:简单注意力和谷歌多头注意力的可视化
文件大小:2.4MB
文件格式:ZIP
更新时间:2024-06-25 16:19:34
系统开源
word源码java
注意可视化
简单注意力和谷歌多头注意力的可视化。
要求
Python
jdk1.8
用法
1.Python版本(仅供简单注意):
python
exec/plot_heatmap.py
--input
xxx.attention
2.Java版本(针对简单注意力和谷歌的多头注意力):
java
-jar
exec/plot_heatmap.jar
然后在
GUI
上选择注意文件。
数据格式
注意文件的名称应以“.attention”扩展名结尾,尤其是在使用
exec/plot_heatmap.jar
时,文件应为
json
格式:
{
"0"
:
{
"source"
:
"
"
,
#
the
source
sentence
(without
and
symbols)
"translation"
:
"
"
,
#
the
target
sentence
(without
and
symbols)
"attentions"
:
[
#
various
attention
results
{
"name"
:
"
"
,
#
a
【文件预览】:
Attention-Visualization-master
----LICENSE(11KB)
----multihead-att-java()
--------HeatmapPanel.java(10KB)
--------MainPanel.java(5KB)
--------MainFrame.java(4KB)
--------ActionLabel.java(3KB)
--------Utils.java(956B)
--------DataObject.java(3KB)
----exec()
--------plot_heatmap.py(3KB)
--------plot_heatmap.jar(105KB)
----README.md(4KB)
----toydata()
--------figures()
--------toy.attention(3.85MB)