Fixes, upgrade, and stuff
This commit is contained in:
@@ -6,7 +6,7 @@ use std::fs;
|
||||
use std::io::{self, BufRead};
|
||||
|
||||
use kickscore as ks;
|
||||
use time::Date;
|
||||
use time::{macros::date, Date};
|
||||
|
||||
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
let reader = fs::File::open("data/nba.csv").map(io::BufReader::new)?;
|
||||
@@ -14,7 +14,8 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
let mut teams = HashSet::new();
|
||||
let mut observations = Vec::new();
|
||||
|
||||
let cutoff = time::date!(2019 - 06 - 01);
|
||||
let cutoff = date!(2019 - 06 - 01);
|
||||
let format = time::format_description::parse("[year]-[month]-[day]").unwrap();
|
||||
|
||||
for line in reader.lines().skip(1) {
|
||||
let line = line?;
|
||||
@@ -22,7 +23,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
|
||||
assert!(data.len() == 5);
|
||||
|
||||
let t = Date::parse(data[0], "%F")?;
|
||||
let t = Date::parse(data[0], &format)?;
|
||||
|
||||
if t > cutoff {
|
||||
break;
|
||||
@@ -73,9 +74,9 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
println!("Probability that CHI beats BOS...");
|
||||
|
||||
let (p_win, _) = model.probabilities(
|
||||
&[&"CHI"],
|
||||
&[&"BOS"],
|
||||
time::date!(1996 - 01 - 01)
|
||||
&["CHI"],
|
||||
&["BOS"],
|
||||
date!(1996 - 01 - 01)
|
||||
.midnight()
|
||||
.assume_utc()
|
||||
.unix_timestamp() as f64,
|
||||
@@ -83,9 +84,9 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
println!(" ... in 1996: {:.2}%", 100.0 * p_win);
|
||||
|
||||
let (p_win, _) = model.probabilities(
|
||||
&[&"CHI"],
|
||||
&[&"BOS"],
|
||||
time::date!(2001 - 01 - 01)
|
||||
&["CHI"],
|
||||
&["BOS"],
|
||||
date!(2001 - 01 - 01)
|
||||
.midnight()
|
||||
.assume_utc()
|
||||
.unix_timestamp() as f64,
|
||||
@@ -93,9 +94,9 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
println!(" ... in 2001: {:.2}%", 100.0 * p_win);
|
||||
|
||||
let (p_win, _) = model.probabilities(
|
||||
&[&"CHI"],
|
||||
&[&"BOS"],
|
||||
time::date!(2020 - 01 - 01)
|
||||
&["CHI"],
|
||||
&["BOS"],
|
||||
date!(2020 - 01 - 01)
|
||||
.midnight()
|
||||
.assume_utc()
|
||||
.unix_timestamp() as f64,
|
||||
|
||||
Reference in New Issue
Block a user