1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

/// # IO NeuroFlow module
/// The module should contain functions and traits for saving
/// and restoring of neural networks from files, and buffers

use std::fs::File;
use std::io::{Read, Write, BufReader};
use serde;
use bincode::{serialize, deserialize_from, Infinite};
use FeedForward;

/// obj: it should be neural network to save.
/// I guess it should be generic type with necessary
/// implementation of some trait
pub fn save<T: serde::Serialize>(obj: &T, file_path: &str){
    let mut file = File::create(file_path).unwrap();
    let encoded: Vec<u8> = serialize(obj, Infinite).unwrap();

    file.write_all(&encoded);
    file.flush();
}

/// Function that load NN from file.
/// Is it possible to return generic type!?
pub fn load<'b, T>(file_path: &'b str) -> T where for<'de> T: serde::Deserialize<'de>{
    let mut content: Vec<u8> = Vec::new();
    let mut file = File::open(file_path).unwrap();
    let mut buf = BufReader::new(file);

    let mut nn: T = deserialize_from(&mut buf, Infinite).unwrap();
    nn
}

/// Future function for saving in JSON string.
/// return: JSON string
pub fn to_json<T: serde::Serialize>(obj: T) {

}

/// Function for deserializing of JSON to NN struct
pub fn from_json(s: &str){

}