UCB-Calculator/script.js at main · 738NGX/UCB-Calculator · GitHub

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
document.addEventListener("DOMContentLoaded", function () {
    const pools = [1, 2, 3];
    let totalDraws = 0;
    let totalUps = 0;
    let upTotals = [0, 0, 0];
    let downTotals = [0, 0, 0];
    let rates = [0, 0, 0];
    const ctx = document.getElementById('poolsChart').getContext('2d');
    const chart = new Chart(ctx, {
        data: {
            labels: ['卡池1', '卡池2', '卡池3'],
            datasets: [{
                label: '出货率',
                type: 'line',
                borderColor: '#e7609e',
                backgroundColor: '#e7609e',
                borderWidth: 0,
                pointBorderWidth: 5,
                pointStyle: 'line',
                pointRadius: 100,
                fill: false,
                data: [0, 0, 0]
            }, {
                label: '置信区间',
                type: 'bar',
                backgroundColor: '#c8c2c6',
                data: [[0, 0], [0, 0], [0, 0]]
            }]
        },
        options: {
            scales: {
                y: {
                    beginAtZero: true
                }
            }
        }
    });

    function updateUCB() {
        pools.forEach((poolNumber, index) => {
            let draws = upTotals[index] + downTotals[index];
            let rate = rates[index];
            let delta = (draws > 0) ? Math.sqrt(Math.log(totalDraws) / (2 * draws)) : 0;
            document.getElementById(`pool${poolNumber}Delta`).textContent = delta.toFixed(4);
            document.getElementById(`pool${poolNumber}UCB`).textContent = (rate + delta).toFixed(4);
            chart.data.datasets[0].data[index] = rate;
            chart.data.datasets[1].data[index][1] = rate + delta;
            chart.data.datasets[1].data[index][0] = rate - delta;
        });
        chart.update();
    }

    pools.forEach(poolNumber => {
        document.getElementById(`drawPool${poolNumber}`).addEventListener('click', function () {
            let upInput = parseInt(document.getElementById(`pool${poolNumber}up`).value, 10) || 0;
            let downInput = parseInt(document.getElementById(`pool${poolNumber}down`).value, 10) || 0;

            upTotals[poolNumber - 1] += upInput;
            downTotals[poolNumber - 1] += downInput;
            let draws = upInput + downInput;
            totalUps += upInput;
            totalDraws += draws;

            document.getElementById(`pool${poolNumber}upTotal`).textContent = upTotals[poolNumber - 1];
            document.getElementById(`pool${poolNumber}downTotal`).textContent = downTotals[poolNumber - 1];
            document.getElementById(`pool${poolNumber}Draws`).textContent = upTotals[poolNumber - 1] + downTotals[poolNumber - 1];
            let rate = (upTotals[poolNumber - 1] > 0) ? upTotals[poolNumber - 1] / (upTotals[poolNumber - 1] + downTotals[poolNumber - 1]) * 100 : 0;
            rates[poolNumber - 1] = rate / 100;
            document.getElementById(`pool${poolNumber}Rate`).textContent = rate.toFixed(2) + '%';

            document.getElementById('totalUps').textContent = totalUps;
            document.getElementById('totalDraws').textContent = totalDraws;
            let totalRate = totalDraws > 0 ? (totalUps / totalDraws) * 100 : 0;
            document.getElementById('totalRate').textContent = totalRate.toFixed(2) + '%';

            updateUCB();

            document.getElementById(`pool${poolNumber}up`).value = "";
            document.getElementById(`pool${poolNumber}down`).value = "";
        });
    });
});