Predict the number of new COVID-19 cases tomorrow

c19.csv

c19.rb

require 'csv'
require 'eps'

# Data for the UK
# Read from here, and definitely rough:
# https://www.bbc.co.uk/news/world-51235105
#
# An example snippet:
# ---------------------
# | day | new | total |
# ---------------------
# | 14  |   0 |     0 |
# | 15  |   1 |     1 |
# | 16  |   4 |     5 |
# | 17  |   0 |     5 |
# | 18  |   1 |     6 |
# ---------------------
FILE_PATH = ENV['C19_FILE_PATH']

text = File.read(FILE_PATH)
csv = CSV.parse(text, headers: true)

rows = csv.map do |raw|
  raw.to_hash.inject({}){|memo,(k,v)| memo[k.to_sym] = v.to_i; memo}
end # https://stackoverflow.com/a/800498/1651458

model = Eps::Model.new(rows, target: :new)
modifier = (rows.last[:day] + 1)
payload = {day: modifier, total: rows.last[:total]}
# I don't know why you have to add the total, but you do.
# Changing it seems to do nothing thus far.

puts "\n"
puts "Making a prediction for day #{payload[:day]}"
puts "Prediction: #{model.predict(payload)}"
puts "\n"