dymk/libmf-rust
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated example [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2021-10-29 12:48:14 -07:00
parent 995bcb2204
commit 59d5297ce2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -197,7 +197,7 @@ fn main() {
.from_reader(file); .from_reader(file);
for (i, record) in rdr.records().enumerate() { for (i, record) in rdr.records().enumerate() {
let row: Row = record.unwrap().deserialize(None).unwrap(); let row: Row = record.unwrap().deserialize(None).unwrap();
let matrix = if i >= 80000 { &mut valid_set } else { &mut train_set }; let matrix = if i < 80000 { &mut train_set } else { &mut valid_set };
matrix.push(row.user_id, row.item_id, row.rating); matrix.push(row.user_id, row.item_id, row.rating);
} }