Basic SIR Model in JS

sir-model.js

function sir (rn, s, i, time, u, p) {
  
  data = {
    S: [s],
    I: [i],
    R: [p-(s+i)]
  }
  
  b = rn*u
  
  for(i = 0; i < time; i++) {
    data.S.push(data.S[i]-(b*data.S[i]*data.I[i])/p) // Susceptible
    data.I.push(data.I[i]+(b*data.S[i]*data.I[i])-(u*data.I[i])/p) // Infected
    data.R.push(data.R[i]+(u*data.I[i])/p) // Recovered
  }
  
}