crfs 0.4.1

Pure Rust port of CRFsuite: a fast implementation of Conditional Random Fields (CRFs)
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

use crfs::train::Trainer;

#[test]
fn test_c1_negative_validation() {
    let mut trainer = Trainer::lbfgs();

    // c1 must be non-negative
    let result = trainer.params_mut().set_c1(-1.0);
    assert!(result.is_err());
    assert_eq!(result.unwrap_err().to_string(), "c1 must be non-negative");

    // c1 = 0.0 should be allowed
    assert!(trainer.params_mut().set_c1(0.0).is_ok());

    // c1 > 0.0 should be allowed
    assert!(trainer.params_mut().set_c1(1.0).is_ok());
}

#[test]
fn test_c2_negative_validation() {
    let mut trainer = Trainer::lbfgs();

    // c2 must be non-negative
    let result = trainer.params_mut().set_c2(-1.0);
    assert!(result.is_err());
    assert_eq!(result.unwrap_err().to_string(), "c2 must be non-negative");

    // c2 = 0.0 should be allowed
    assert!(trainer.params_mut().set_c2(0.0).is_ok());

    // c2 > 0.0 should be allowed
    assert!(trainer.params_mut().set_c2(1.0).is_ok());
}

#[test]
fn test_epsilon_validation() {
    let mut trainer = Trainer::lbfgs();

    // epsilon must be non-negative
    assert!(trainer.params_mut().set_epsilon(0.0).is_ok());

    // epsilon < 0.0 should fail
    let result = trainer.params_mut().set_epsilon(-0.001);
    assert!(result.is_err());
    assert_eq!(
        result.unwrap_err().to_string(),
        "epsilon must be non-negative"
    );

    // epsilon > 0.0 should be allowed
    assert!(trainer.params_mut().set_epsilon(0.001).is_ok());
    assert!(trainer.params_mut().set_epsilon(1e-5).is_ok());
}

#[test]
fn test_invalid_parameter_values() {
    let mut trainer = Trainer::lbfgs();

    assert!(trainer.params_mut().set_num_memories(0).is_err());
    assert!(trainer.params_mut().set_max_iterations(0).is_err());

    let mut l2sgd_trainer = Trainer::l2sgd();
    assert!(l2sgd_trainer.params_mut().set_period(0).is_err());
    assert!(l2sgd_trainer.params_mut().set_delta(0.0).is_err());
}