Skip to content

Commit 0187b45

Browse files
authored
Add Serialize trait to core data structures (#370)
1 parent 163f031 commit 0187b45

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/inference.rs

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
use burn::backend::ndarray::NdArrayDevice;
22
use itertools::izip;
3+
use serde::Serialize;
34
use std::collections::HashMap;
45
use std::ops::{Add, Sub};
56

@@ -72,7 +73,7 @@ pub fn current_retrievability(state: MemoryState, days_elapsed: f32, decay: f32)
7273
(days_elapsed / state.stability * factor + 1.0).powf(-decay)
7374
}
7475

75-
#[derive(Debug, PartialEq, Clone, Copy)]
76+
#[derive(Debug, PartialEq, Clone, Copy, Serialize)]
7677
pub struct MemoryState {
7778
pub stability: f32,
7879
pub difficulty: f32,
@@ -611,15 +612,15 @@ pub struct ModelEvaluation {
611612
pub rmse_bins: f32,
612613
}
613614

614-
#[derive(Debug, Clone, PartialEq)]
615+
#[derive(Debug, Clone, PartialEq, Serialize)]
615616
pub struct NextStates {
616617
pub again: ItemState,
617618
pub hard: ItemState,
618619
pub good: ItemState,
619620
pub easy: ItemState,
620621
}
621622

622-
#[derive(Debug, PartialEq, Clone)]
623+
#[derive(Debug, PartialEq, Clone, Serialize)]
623624
pub struct ItemState {
624625
pub memory: MemoryState,
625626
pub interval: f32,

0 commit comments

Comments
 (0)