GitHub/Anki
1
0
Fork 0
mirror of https://github.com/ankitects/anki.git synced 2025-09-18 22:12:21 -04:00
Code Issues Projects Releases Packages 1 Wiki Activity Actions

Show review count with FSRS evaluation results (#3165)

Browse source
This commit is contained in:
Abdo 2024-04-25 10:10:41 +03:00 committed by GitHub
parent ee8683587a
commit b7e782c7e1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 18 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
proto/anki/scheduler.proto
View file

@ -434,6 +434,7 @@ message EvaluateWeightsRequest {
message EvaluateWeightsResponse { message EvaluateWeightsResponse {
float log_loss = 1; float log_loss = 1;
float rmse_bins = 2; float rmse_bins = 2;
uint32 review_count = 3;
} }
message ComputeMemoryStateResponse { message ComputeMemoryStateResponse {

9
rslib/src/scheduler/fsrs/weights.rs
View file

@ -152,7 +152,7 @@ impl Collection {
weights: &Weights, weights: &Weights,
search: &str, search: &str,
ignore_revlogs_before: TimestampMillis, ignore_revlogs_before: TimestampMillis,
) -> Result<ModelEvaluation> { ) -> Result<(u32, ModelEvaluation)> {
let timing = self.timing_today()?; let timing = self.timing_today()?;
let mut anki_progress = self.new_progress_handler::<ComputeWeightsProgress>(); let mut anki_progress = self.new_progress_handler::<ComputeWeightsProgress>();
let guard = self.search_cards_into_table(search, SortMode::NoOrder)?; let guard = self.search_cards_into_table(search, SortMode::NoOrder)?;
@ -164,14 +164,17 @@ impl Collection {
fsrs_items_for_training(revlogs, timing.next_day_at, ignore_revlogs_before); fsrs_items_for_training(revlogs, timing.next_day_at, ignore_revlogs_before);
anki_progress.state.reviews = review_count as u32; anki_progress.state.reviews = review_count as u32;
let fsrs = FSRS::new(Some(weights))?; let fsrs = FSRS::new(Some(weights))?;
Ok(fsrs.evaluate(items, |ip| { Ok((
review_count as u32,
fsrs.evaluate(items, |ip| {
anki_progress anki_progress
.update(false, |p| { .update(false, |p| {
p.total_iterations = ip.total as u32; p.total_iterations = ip.total as u32;
p.current_iteration = ip.current as u32; p.current_iteration = ip.current as u32;
}) })
.is_ok() .is_ok()
})?) })?,
))
} }
} }

3
rslib/src/scheduler/service/mod.rs
View file

@ -287,7 +287,7 @@ impl crate::services::SchedulerService for Collection {
&mut self, &mut self,
input: scheduler::EvaluateWeightsRequest, input: scheduler::EvaluateWeightsRequest,
) -> Result<scheduler::EvaluateWeightsResponse> { ) -> Result<scheduler::EvaluateWeightsResponse> {
let ret = self.evaluate_weights( let (review_count, ret) = self.evaluate_weights(
&input.weights, &input.weights,
&input.search, &input.search,
input.ignore_revlogs_before_ms.into(), input.ignore_revlogs_before_ms.into(),
@ -295,6 +295,7 @@ impl crate::services::SchedulerService for Collection {
Ok(scheduler::EvaluateWeightsResponse { Ok(scheduler::EvaluateWeightsResponse {
log_loss: ret.log_loss, log_loss: ret.log_loss,
rmse_bins: ret.rmse_bins, rmse_bins: ret.rmse_bins,
review_count,
}) })
} }

4
ts/routes/deck-options/FsrsOptions.svelte
View file

@ -172,7 +172,9 @@ License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
setTimeout( setTimeout(
() => () =>
alert( alert(
`Log loss: ${resp.logLoss.toFixed(4)}, RMSE(bins): ${( `Reviews: ${
resp.reviewCount
}, Log loss: ${resp.logLoss.toFixed(4)}, RMSE(bins): ${(
resp.rmseBins * 100 resp.rmseBins * 100
).toFixed(2)}%. ${tr.deckConfigSmallerIsBetter()}`, ).toFixed(2)}%. ${tr.deckConfigSmallerIsBetter()}`,
), ),