Added more metrics

This commit is contained in:
Andrew Kane
2021-07-26 14:06:36 -07:00
parent a095755bb9
commit cc0024a9f9
4 changed files with 76 additions and 2 deletions

View File

@@ -77,18 +77,48 @@ model.cv(&data, 5);
## Metrics
Calculate RMSE
Calculate RMSE (for real-valued MF)
```rust
model.rmse(&data);
```
Calculate MAE
Calculate MAE (for real-valued MF)
```rust
model.mae(&data);
```
Calculate generalized KL-divergence (for non-negative real-valued MF)
```rust
model.gkl(&data);
```
Calculate logarithmic loss (for binary MF)
```rust
model.logloss(&data);
```
Calculate accuracy (for binary MF)
```rust
model.accuracy(&data);
```
Calculate MPR (for one-class MF)
```rust
model.mpr(&data, transpose);
```
Calculate AUC (for one-class MF)
```rust
model.auc(&data, transpose);
```
## Parameters
Set parameters - default values below