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 3d184073d7
commit eb597475c5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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();