multi-skill 0.1.2

Skill estimation systems for multiplayer competitions
Documentation 
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
83
84
85

extern crate multi_skill;

use multi_skill::data_processing::{get_dataset_by_name, write_slice_to_file, ContestSummary};
use multi_skill::experiment_config::Experiment;
use multi_skill::summary::print_ratings;
use multi_skill::systems::{get_rating_system_by_name, simulate_contest};
use std::collections::HashMap;

fn get_experiment_from_args(args: &[String]) -> Experiment {
    if args[1] == "file:" {
        return Experiment::from_file(&args[2]);
    }

    let system = get_rating_system_by_name(&args[1]).unwrap();
    let dataset = get_dataset_by_name(&args[2]).unwrap();
    let max_contests = args
        .get(3)
        .and_then(|s| s.parse().ok())
        .unwrap_or(usize::MAX);

    Experiment {
        max_contests,
        mu_noob: 1500.,
        sig_noob: 350.,
        system,
        dataset,
    }
}

/// Simulates the entire history of Codeforces
fn main() {
    // Parse arguments, prepare rating system and datasets
    let args: Vec<String> = std::env::args().collect();
    if args.len() != 3 && args.len() != 4 {
        eprintln!("Usage: {} system_name dataset_name [num_contests]", args[0]);
        return;
    }
    let ex = get_experiment_from_args(&args);

    // Simulate the contests and rating updates
    let mut players = HashMap::new();
    let mut summaries = Vec::with_capacity(ex.dataset.len().min(ex.max_contests));
    let mut last_contest_time = 0;
    for (index, contest) in ex.dataset.iter().enumerate().take(ex.max_contests) {
        println!(
            "Processing{:6} contestants in{:5}th contest with wt={}: {}",
            contest.standings.len(),
            index,
            contest.weight,
            contest.name
        );
        simulate_contest(
            &mut players,
            &contest,
            &*ex.system,
            ex.mu_noob,
            ex.sig_noob,
            index,
        );
        summaries.push(ContestSummary::new(&contest));
        last_contest_time = contest.time_seconds;
    }
    let six_months_ago = last_contest_time.saturating_sub(183 * 86_400);
    let dir = std::path::PathBuf::from("../data/output");
    std::fs::create_dir_all(&dir.join("players")).expect("Could not create directory");

    // Print contest histories of top players to data/output/players/{handle}.csv
    for (handle, player) in &players {
        let player = player.borrow();
        // let last_event = player.event_history.last().expect("Empty history");

        // if last_event.display_rating >= 2700 && player.update_time > six_months_ago
        if true {
            let player_file = dir.join(format!("players/{}.csv", handle));
            write_slice_to_file(&player.event_history, &player_file);
        }
    }

    // Print ratings list to data/codeforces/CFratings.txt
    print_ratings(&players, six_months_ago, &dir);

    // Write contest summaries to data/codeforces/summaries.csv
    let summary_file = dir.join("all_contests.csv");
    write_slice_to_file(&summaries, &summary_file);
}