Added disk-level training

This commit is contained in:
Andrew Kane
2021-07-26 15:14:21 -07:00
parent 27cb6fa2a9
commit 7cbcf78a95
4 changed files with 62 additions and 0 deletions

View File

@@ -160,6 +160,24 @@ Calculate AUC (for one-class MF)
model.auc(&data, transpose);
```
## Disk-Level Training
Train directly from files
```ruby
model.fit_disk("train.txt")
model.fit_eval_disk("train.txt", "validate.txt")
model.cv_disk("train.txt")
```
Data should be in the format `row_index column_index value`:
```txt
0 0 5.0
0 2 3.5
1 1 4.0
```
## Reference
Specify the initial capacity for a matrix