diff-priv 0.1.0

k-anonymity, (c,l)-diversity and ε-differential privacy framework
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
86
87
88
89
90

use crate::test::environment::Datasets;
use csv::Writer;

const MUELLER_HEADERS: [&str; 16] = [
    "uuid",
    "disclosure_risk",
    "id",
    "gender",
    "age",
    "run_id",
    "running",
    "speed",
    "duration",
    "distance",
    "side",
    "bout",
    "freq",
    "walk_ratio",
    "start",
    "end",
];

const ADULT_LARGE_HEADERS: [&str; 15] = [
    "uuid",
    "disclosure_risk",
    "timestamp",
    "age",
    "fnlwgt",
    "education_num",
    "capital_gain",
    "capital_loss",
    "hours_per_week",
    "education",
    "marital_status",
    "workclass",
    "native_country",
    "occupation",
    "class",
];

const ADULT_HEADERS: [&str; 10] = [
    "uuid",
    "disclosure_risk",
    "timestamp",
    "age",
    "fnlwgt",
    "education_num",
    "capital_gain",
    "capital_loss",
    "hours_per_week",
    "class",
];

pub struct CsvExporter {
    data: Vec<Vec<String>>,
    pub path: String,
    dataset: Datasets,
}

impl CsvExporter {
    pub fn new(path: String, dataset: Datasets) -> Self {
        Self {
            path,
            data: vec![],
            dataset,
        }
    }

    pub fn export(&self) -> Result<(), Box<dyn std::error::Error>> {
        println!("Exporting to {}", self.path);
        let mut writer = Writer::from_path(&self.path)?;
        // write header
        match self.dataset {
            Datasets::Adult(_) => writer.write_record(&ADULT_HEADERS)?,
            Datasets::AdultLarge(_) => writer.write_record(&ADULT_LARGE_HEADERS)?,
            Datasets::Mueller(_) => writer.write_record(&MUELLER_HEADERS)?,
        }

        for record in &self.data {
            writer.write_record(record)?
        }

        writer.flush()?;
        Ok(())
    }

    pub fn add(&mut self, value: Vec<String>) {
        self.data.push(value)
    }
}