Test RealKL loss [skip ci]
This commit is contained in:
@@ -146,6 +146,12 @@ mod tests {
|
||||
assert_eq!(model.bias(), 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_loss_real_kl() {
|
||||
let data = generate_data();
|
||||
assert!(Model::params().loss(Loss::RealKL).nmf(true).quiet(true).fit(&data).is_ok());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_save_load() {
|
||||
let data = generate_data();
|
||||
|
||||
Reference in New Issue
Block a user