Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 9,041 Bytes
e2ef657 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 |
import {D3Sel} from "../etc/Util";
import * as d3 from "d3";
export function createStaticSkeleton(base: D3Sel) {
// ---- TEMPLATE -----
// language=HTML
function template() {
return `
<!-- <div id="attention-vis">-->
<div>
<div id="model-selection">
<label for="model-options"
style="width:150px;margin-bottom: 5px;">Select
model</label><select id="model-option-selector"
name="model-options">
</select>
</div>
</div>
<div id="sentence-input">
<form>
<div class="form-group">
<label for="form-sentence-a"
style="width: 150px; display: inline-block;vertical-align: middle;">Input
Sentence</label>
<input id="form-sentence-a" type="text"
name="sent-a-input"
placeholder="Enter new sentence to analyze"
value="The girl ran to a local pub to escape the din of her city."
style="width: calc(100% - 240px); display: inline-block;vertical-align: middle;"
>
<button class="btn btn-primary" id="update-sentence"
type="button"
style="width:70px;display: inline-block;padding: 3px 3px;">
Update
</button>
</div>
</form>
<!-- <div class="padding"></div>-->
</div>
<div>
<div style="display: inline-block; width:150px;"><b>Filters</b>
</div>
<div id="cls-toggle" style="display: inline-block;">
<!-- <div class="input-description">-->
Hide Special Tokens
<!-- </div>-->
<label class="switch" style="vertical-align: middle;">
<input type="checkbox" checked="checked">
<span class="short-slider round"></span>
</label>
</div>
<div style="margin-left: 25px;display: inline-block;">
Show top <span id="my-range-value">70</span>% of att:
</div>
<div class="slide-container"
style="display: inline-block;margin-left: 5px;width:calc(100% - 550px);">
<!-- <label for="my-range">-->
<!-- </label>-->
<input
type="range" min="0" max="100" value="70"
class="slider" id="my-range"
style="vertical-align: middle;display: inline-block;">
</div>
</div>
<div id="layer-selection" style="margin-top: 10px;">
<div class="input-description"
style="display: inline-block; width: 150px;vertical-align: top;">
Layer
</div>
<div class="layer-select btn-group btn-group-toggle"
data-toggle="buttons" id="layer-select"
style="vertical-align: top;">
</div>
</div>
<div id="selected-head-display">
<div class="input-description" style="display: inline-block; width: 150px;vertical-align: top;">
Selected heads:
</div>
<div id="selected-heads" style="display:inline-block;vertical-align: top;">
</div>
</div>
<div id="connector-container">
<div id="connector-controls">
<div class="left-control-half">
<!-- <div id="layer-selection">-->
<!-- <div class="input-description">-->
<!-- Layer:-->
<!-- </div>-->
<!-- <div class="layer-select btn-group btn-group-toggle"-->
<!-- data-toggle="buttons" id="layer-select">-->
<!-- </div>-->
<!-- </div>-->
</div>
<div class="right-control-half head-control">
<!-- <div id="selected-head-display">-->
<!-- <div class="input-description">-->
<!-- Selected heads:-->
<!-- </div>-->
<!-- <div id="selected-heads">-->
<!-- </div>-->
<!-- </div>-->
<div class="select-input" id="head-all-or-none">
<button id="select-all-heads">Select all heads
</button>
<button id="select-no-heads">Unselect all heads
</button>
</div>
<div id="usage-info">
<p>
You focus on one token by <b>click</b>.
You can mask any token by <b>double click</b>.
</p>
<p>
You can select and de-select a head by a
<b>click</b>
on the heatmap columns
</p>
</div>
</div>
</div>
<div id="atn-container" class="text-center">
<div id="head-info-box"
class="mat-hover-display text-center"
style="width: 70px; height: 30px; visibility: hidden; left: 929px; top: 670px; border-radius: 8px 8px 8px 1px;">
</div>
<svg id="left-att-heads" width="182" height="442"></svg>
<div id="left-tokens"></div>
<svg id="atn-display" height="442" width="200"></svg>
<div id="right-tokens"></div>
<svg id="right-att-heads" width="182" height="442"></svg>
</div>
<div id="vis-break"></div>
</div>
<!-- </div>-->
`
}
base.html(template);
// ---- TEMPLATE.end -----
/**
* Top level sections
*/
const sentenceInput = base.select("#sentence-input");
const connectorContainer = base.select('#connector-container');
const atnControls = base.select('#connector-controls');
const atnContainer = base.select('#atn-container');
/**
* Sentence Input
*/
const sentenceA = base.select('#form-sentence-a');
const formButton = base.select('#update-sentence');
/**
* Connector Controls
*/
const modelSelector = base.select('#model-option-selector')
const threshSlider = base.select('#my-range');
const layerCheckboxes = base.select('#layer-select');
const clsToggle = base.select('#cls-toggle');
const selectedHeads = base.select('#selected-heads');
const headSelectAll = base.select('#select-all-heads');
const headSelectNone = base.select('#select-no-heads');
/**
* For main attention vis
*/
const headInfoBox = base.select('#head-info-box');
const headBoxLeft = base.select('#left-att-heads');
const tokensLeft = base.select('#left-tokens');
const atnDisplay = base.select('#atn-display');
const tokensRight = base.select('#right-tokens');
const headBoxRight = base.select('#right-att-heads');
/**
* Return an object that provides handles to the important parts here
*/
return {
body: d3.select('body'),
atnContainer: atnContainer,
atnDisplay: atnDisplay,
atnHeads: {
left: headBoxLeft,
right: headBoxRight,
headInfo: headInfoBox
},
form: {
sentenceA: sentenceA,
button: formButton
},
tokens: {
left: tokensLeft,
right: tokensRight
},
modelSelector: modelSelector,
clsToggle: clsToggle,
layerCheckboxes: layerCheckboxes,
selectedHeads: selectedHeads,
headSelectAll: headSelectAll,
headSelectNone: headSelectNone,
threshSlider: threshSlider,
}
}
|