Changed fit and fit_eval to return Result

This commit is contained in:
Andrew Kane
2021-10-16 17:43:37 -07:00
parent af64e749cd
commit 26160cfb9a
6 changed files with 37 additions and 25 deletions

View File

@@ -26,7 +26,7 @@ data.push(1, 1, 4.0);
Fit a model
```rust
let model = libmf::Model::params().fit(&data);
let model = libmf::Model::params().fit(&data).unwrap();
```
Make predictions
@@ -63,7 +63,7 @@ let model = libmf::Model::load("model.txt").unwrap();
Pass a validation set
```rust
let model = libmf::Model::params().fit_eval(&train_set, &eval_set);
let model = libmf::Model::params().fit_eval(&train_set, &eval_set).unwrap();
```
## Cross-Validation
@@ -179,7 +179,7 @@ let mut data = libmf::Matrix::with_capacity(3);
Use
```rust
let model = libmf::Model::params().factors(20).fit(&data);
let model = libmf::Model::params().factors(20).fit(&data).unwrap();
```
instead of