mervenoyan's picture
commit files to HF hub
d8760c5
console.clear()
d3.select('body').selectAppend('div.tooltip.tooltip-hidden')
var colors = ["#c0584b", "#649398"]
d3.loadData('https://roadtolarissa.com/colab/gender-over-time-colab/processed_vocab.json', (err, res) => {
window.vocab = res[0]
d3.select('#graph').html('').datum(jsData).each(drawSentence)
})
async function drawSentence({s0, s1, tidyCSV, minYear}, i){
var tidy = d3.csvParse(jsData.tidyCSV)
console.log(minYear)
tidy.forEach(d => {
d.year = minYear + +d.year_index
d.i = +d.token_index
d.e0 = +d.e0
d.e1 = +d.e1
d.mean = d.e0 + d.e1
d.dif = d.e0 - d.e1
})
var sel = d3.select(this).st({marginRight: 20})
sel.append('div').st({color: colors[0]}).text(s0)
sel.append('div').st({color: colors[1]}).text(s1)
var c = d3.conventions({
sel,
height: 300,
width: 300,
margin: {left: 0, bottom: 50, right: 60}
})
c.svg.append('rect')
.at({width: c.width, height: c.height/2, opacity: .1, fill: colors[0]})
c.svg.append('rect')
.at({width: c.width, height: c.height/2, opacity: .1, fill: colors[1], y: c.height/2})
console.log(tidy[0])
c.x.domain(d3.extent(tidy, d => d.year))
var minY = d3.min(d3.extent(tidy, d => d.dif), Math.abs)
var maxY = d3.max(d3.extent(tidy, d => d.dif), Math.abs)
c.y.domain([-minY, minY])
c.y.domain([-maxY, maxY])
// c.y.domain([-4.5, 4.5])
c.xAxis.tickFormat(d => d)
c.yAxis.ticks(5)
d3.drawAxis(c)
var byToken = d3.nestBy(tidy, d => d.i)
byToken.forEach(d => {
d.endY = c.y(_.last(d).dif)
d.str = vocab[+d.key].replace('▁', '')
d.displayLabel = true
d.mean = d3.sum(d, e => e.mean)
})
console.log(tidy[0])
d3.nestBy(_.sortBy(byToken, d => -d.mean), d => Math.round(d.endY/12))
.forEach(d => d.forEach((e, i) => e.displayLabel = !i))
var line = d3.line()
.x(d => c.x(d.year))
.y(d => c.y(d.dif))
var tokenSel = c.svg.appendMany('g', byToken)
// .call(d3.attachTooltip)
.on('mouseover', function(){
tokenSel.classed('active', 0)
d3.select(this)
.raise()
.classed('active', 1)
})
c.svg.on('mouseleave', function(){
tokenSel.classed('active', 0)
})
tokenSel.append('text')
.text(d => d.str)
.translate(d => [c.width + 2, d.endY])
.at({fontSize: 10, dy: '.33em', fill: (d, i) => d.displayLabel ? '#999' : 'rgba(0,0,0,0)'})
tokenSel.append('path')
.at({
d: line,
stroke: '#000',
opacity: .2,
fill: 'none',
})
}