Fix/set pretrain_only by fsrs_items (#3051)

* Fix/set pretrain_only and insufficient reviews based on fsrs_items

* use review_count for FsrsInsufficientReviews

* Update weights.rs

* Update weights.rs
This commit is contained in:
Jarrett Ye 2024-03-05 15:54:34 +08:00 committed by GitHub
parent 191dd7dc64
commit 5faaff7d56

View file

@ -94,7 +94,7 @@ impl Collection {
});
let fsrs = FSRS::new(Some(current_weights))?;
let mut weights =
fsrs.compute_parameters(items.clone(), revlogs.len() < 1000, Some(progress2))?;
fsrs.compute_parameters(items.clone(), fsrs_items < 1000, Some(progress2))?;
let metrics = fsrs.universal_metrics(items, &weights, |_| true)?;
if metrics.0 < metrics.1 {
weights = current_weights.to_vec();