//TensorFlow.js// Video: https://youtu.be/tIXDik5SGsI disableFriendlyErrors = true; let cnv,dtsel; let x_vals = []; //days let y_vals = []; //cases let Country = "World"; let Day = 30; let nextDay = 1; //next day value for prediction let dcases,ddeaths,dncases,dndeaths,deg; let coffy = []; //coefficients deg = 3; //degree of polynomial //low degree->under fitting, high degree overfitting. //higher degree let dataT =[]; //datasets let dd =1; //datasets index 1-4 const optimizer = tf.train.adam(learnrate=0.5); //list from w3scools.com, modified let countries = ["World","Afghanistan","Albania","Algeria","Andorra","Angola","Anguilla","Antigua & Barbuda","Argentina","Armenia","Aruba","Australia","Austria","Azerbaijan","Bahamas","Bahrain","Bangladesh","Barbados","Belarus","Belgium","Belize","Benin","Bermuda","Bhutan","Bolivia","Bosnia & Herzegovina","Botswana","Brazil","British Virgin Islands","Brunei","Bulgaria","Burkina Faso","Burundi","Cambodia","Cameroon","Canada","Cape Verde","Cayman Islands","Central Arfrican Republic","Chad","Chile","China","Colombia","Congo","Cook Islands","Costa Rica","Cote D Ivoire","Croatia","Cuba","Curacao","Cyprus","Czech Republic","Denmark","Djibouti","Dominica","Dominican Republic","Ecuador","Egypt","El Salvador","Equatorial Guinea","Eritrea","Estonia","Ethiopia","Falkland Islands","Faroe Islands","Fiji","Finland","France","French Polynesia","French West Indies","Gabon","Gambia","Georgia","Germany","Ghana","Gibraltar","Greece","Greenland","Grenada","Guam","Guatemala","Guernsey","Guinea","Guinea Bissau","Guyana","Haiti","Honduras","Hong Kong","Hungary","Iceland","India","Indonesia","Iran","Iraq","Ireland","Isle of Man","Israel","Italy","Jamaica","Japan","Jersey","Jordan","Kazakhstan","Kenya","Kiribati","Kosovo","Kuwait","Kyrgyzstan","Laos","Latvia","Lebanon","Lesotho","Liberia","Libya","Liechtenstein","Lithuania","Luxembourg","Macau","Macedonia","Madagascar","Malawi","Malaysia","Maldives","Mali","Malta","Marshall Islands","Mauritania","Mauritius","Mexico","Micronesia","Moldova","Monaco","Mongolia","Montenegro","Montserrat","Morocco","Mozambique","Myanmar","Namibia","Nauro","Nepal","Netherlands","Netherlands Antilles","New Caledonia","New Zealand","Nicaragua","Niger","Nigeria","North Korea","Norway","Oman","Pakistan","Palau","Palestine","Panama","Papua New Guinea","Paraguay","Peru","Philippines","Poland","Portugal","Puerto Rico","Qatar","Reunion","Romania","Russia","Rwanda","Saint Pierre & Miquelon","Samoa","San Marino","Sao Tome and Principe","Saudi Arabia","Senegal","Serbia","Seychelles","Sierra Leone","Singapore","Slovakia","Slovenia","Solomon Islands","Somalia","South Africa","South Korea","South Sudan","Spain","Sri Lanka","St Kitts & Nevis","St Lucia","St Vincent","Sudan","Suriname","Swaziland","Sweden","Switzerland","Syria","Taiwan","Tajikistan","Tanzania","Thailand","Timor L'Este","Togo","Tonga","Trinidad & Tobago","Tunisia","Turkey","Turkmenistan","Turks & Caicos","Tuvalu","Uganda","Ukraine","United Arab Emirates","United Kingdom","United States","Uruguay","Uzbekistan","Vanuatu","Vatican City","Venezuela","Vietnam","Virgin Islands (US)","Yemen","Zambia","Zimbabwe"]; //loading some unloaded stuff function preload(){ dataT[1] = loadTable('https://covid.ourworldindata.org/data/ecdc/total_cases.csv','csv', 'header'); dataT[2] = loadTable('https://covid.ourworldindata.org/data/ecdc/total_deaths.csv','csv','header'); dataT[3] = loadTable('https://covid.ourworldindata.org/data/ecdc/new_cases.csv','csv','header'); dataT[4] = loadTable('https://covid.ourworldindata.org/data/ecdc/new_deaths.csv','csv','header'); } //preparing data (array) function dataf(d=dd, country=Country, day=Day){ //creating an array of last 30 values x_vals = [];y_vals =[]; data=dataT[dd].getColumn(country).slice(-day); // for(let i of data){ // print(i) // } for(let i =0;ideg+1){ coffy[i].dispose();} coffy[i] = tf.variable( tf.scalar(0)); } } //run once function setup() { cnv = createCanvas(windowWidth, 400); cnv.parent('canva'); dtsel = createSelect(); dtsel.option('1. Total cases'); dtsel.option('2. Total deaths'); dtsel.option('3. New cases'); dtsel.option('4. New deaths'); dtsel.selected('1. Total cases'); dtsel.changed(()=>dataf( dd = parseInt(dtsel.value()[0]))); createSpan(` `); print('wait ⏳ for the values to be fully stable'); print('Enter your country name to see prediction for for tommorow'); createElement('br'); createButton('History 30/60 days') .mousePressed( ()=>{ if(Day==60) Day=30; else Day=60; dataf(); }) createButton('Linear/Polynomial') .mousePressed( ()=>{ if(deg ==1) deg=3; else {deg=1; print('Linear? compare loss')} }) dataf(); initcoffy(); } function loss(pred, labels) { //RMS error let los= pred.sub(labels) .square() .mean(); fill(255,170,0); strokeWeight(0) losVal = los.dataSync()[0]; text(`Loss: ${losVal}`,130,10); if(losVal>1000){initcoffy()} return los } //Astrologer predicts ys (cases) bt 0-1 function predict(x, n=deg) { const xs = tf.tensor1d(x); // y = ax^n + bx^(n-1) +...+ const ys = coffy[0]; for(let i=1;i<=n;i++){ ys = xs.pow(tf.scalar(i)).mul(coffy[i]).add(ys); } return ys; } //now u r in a infinite loop function draw() { background(0) tf.tidy(() => { if (x_vals.length) { const ys = tf.tensor1d(y_vals); optimizer.minimize( () =>loss( predict(x_vals,deg), ys )); }}); //🖌️🎨 lets draw something textSize(13); stroke(255); strokeWeight(8); //Dots for (let i = 0; i < x_vals.length; i++) { let px = x_vals[i]*(width-20); let py = 20+ (1-y_vals[i])*(height-20); stroke( y_vals[i]*255, 255-x_vals[i]*255, 200); point(px, py); } const linespaceX = []; for (let x = -1; x <= 1.2; x += 0.05) { linespaceX.push(x); } const ys = tf.tidy( () => predict(linespaceX,deg)); let linespaceY = ys.dataSync(); ys.dispose(); beginShape(); noFill(); strokeWeight(2); for (let i = 0; i < linespaceX.length; i++) { let x = map(linespaceX[i], 0, 1, 0, width-20); let y = map(linespaceY[i], 0, 1, height, 20); vertex(x, y); } endShape(); fill(0,255,0) strokeWeight(0); text(`active tensors: ${tf.memory().numTensors}`,5,10); fill(255); text(`Last update: ${dataT[dd].getColumn('date').slice(-1)[0]}\n${dtsel.value().substr(3)}: ${data[data.length-1]}`,5,25); fill(255,150,220); text(`prediction for next day\n${parseInt(tf.tidy(()=>predict([1+(x_vals[1])*nextDay]).dataSync()*data[data.length-1]))} ± ${Math.hypot(tf.tidy(()=>1-predict([1]).dataSync()))*100}%`,5,58); fill('#ffff00') text('days----->',width/2-20,height-2); push(); fill('#ffff00'); translate(8,height/2+50); rotate(-PI/2) text('cases--->',0,0); pop(); } //from W3schools, modified function clist(inp,arr) { var currentFocus; inp.addEventListener("input", function(e) { var a, b, i, val = this.value; closeAllLists(); if (!val) { return false;} currentFocus = -1; a = document.createElement("DIV"); a.setAttribute("id", this.id + "autocomplete-list"); a.setAttribute("class", "autocomplete-items"); this.parentNode.appendChild(a); for (i = 0; i < arr.length; i++) { if (arr[i].substr(0, val.length).toUpperCase() == val.toUpperCase()) { b = document.createElement("DIV"); b.innerHTML = "" + arr[i].substr(0, val.length) + ""; b.innerHTML += arr[i].substr(val.length); b.innerHTML += ""; b.addEventListener("click", function(e) { inp.value = this.getElementsByTagName("input")[0].value; Country = inp.value; dataf(); closeAllLists(); }); a.appendChild(b); } } }); inp.addEventListener("keydown", function(e) { var x = document.getElementById(this.id + "autocomplete-list"); if (x) x = x.getElementsByTagName("div"); if (e.keyCode == 40) { currentFocus++; addActive(x); } else if (e.keyCode == 38) { currentFocus--; addActive(x); } else if (e.keyCode == 13){ e.preventDefault(); if (currentFocus > -1) { if (x) x[currentFocus].click(); } } }); function addActive(x) { if (!x) return false; removeActive(x); if (currentFocus >= x.length) currentFocus = 0; if (currentFocus < 0) currentFocus = (x.length - 1); x[currentFocus].classList.add("autocomplete-active"); } function removeActive(x) { for (var i = 0; i < x.length; i++) { x[i].classList.remove("autocomplete-active"); } } function closeAllLists(elmnt) { var x = document.getElementsByClassName("autocomplete-items"); for (var i = 0; i < x.length; i++) { if (elmnt != x[i] && elmnt != inp) { x[i].parentNode.removeChild(x[i]); } } } document.addEventListener("click", function (e) { closeAllLists(e.target); }); }

Covid19 Prediction || TensorFlow.js