lawkit-core 2.1.0

Core library for statistical law analysis with international number support
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

pub mod benford;
pub mod normal;
pub mod pareto;
pub mod poisson;
pub mod zipf;

pub use benford::*;
pub use normal::*;
pub use pareto::*;
pub use poisson::*;
pub use zipf::*;

use crate::error::Result;
use rand::rngs::StdRng;
use rand::SeedableRng;

#[derive(Debug, Clone)]
pub struct GenerateConfig {
    pub samples: usize,
    pub seed: Option<u64>,
    pub output_format: String,
    pub fraud_rate: f64,
}

impl GenerateConfig {
    pub fn new(samples: usize) -> Self {
        Self {
            samples,
            seed: None,
            output_format: "text".to_string(),
            fraud_rate: 0.0,
        }
    }

    pub fn with_seed(mut self, seed: u64) -> Self {
        self.seed = Some(seed);
        self
    }

    pub fn with_fraud_rate(mut self, rate: f64) -> Self {
        self.fraud_rate = rate.clamp(0.0, 1.0);
        self
    }

    pub fn create_rng(&self) -> StdRng {
        match self.seed {
            Some(seed) => StdRng::seed_from_u64(seed),
            None => StdRng::from_entropy(),
        }
    }
}

pub trait DataGenerator {
    type Output;

    fn generate(&self, config: &GenerateConfig) -> Result<Self::Output>;
}