Test RealKL loss [skip ci]

This commit is contained in:
Andrew Kane
2021-10-16 19:34:32 -07:00
parent 40accfa870
commit 1da4ac9c3a

View File

@@ -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();