work/smartcore
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: bug in dataset serializer

Browse Source
This commit is contained in:
Volodymyr Orlov
2020-09-14 13:45:58 -07:00
parent 7c013ddc35
commit e0e4791cbc
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/dataset/mod.rs
+2
View File
@@ -12,6 +12,7 @@ use std::io;
use std::io::prelude::*; use std::io::prelude::*;
/// Dataset /// Dataset
#[derive(Debug)]
pub struct Dataset<X, Y> { pub struct Dataset<X, Y> {
/// data in one-dimensional array. /// data in one-dimensional array.
pub data: Vec<X>, pub data: Vec<X>,
@@ -102,6 +103,7 @@ pub(crate) fn deserialize_data(
for _ in 0..(num_samples) { for _ in 0..(num_samples) {
buffer.copy_from_slice(&bytes[c..(c + 4)]); buffer.copy_from_slice(&bytes[c..(c + 4)]);
y.push(f32::from_bits(u32::from_le_bytes(buffer))); y.push(f32::from_bits(u32::from_le_bytes(buffer)));
c += 4;
} }
Ok((x, y, num_samples, num_features)) Ok((x, y, num_samples, num_features))