From 59d5297ce24aafd3d4226b2608c88e488b234912 Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Fri, 29 Oct 2021 12:48:14 -0700 Subject: [PATCH] Updated example [skip ci] --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 83ad15b..bf58074 100644 --- a/README.md +++ b/README.md @@ -197,7 +197,7 @@ fn main() { .from_reader(file); for (i, record) in rdr.records().enumerate() { 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); }