93 lines
3.3 KiB
TypeScript
93 lines
3.3 KiB
TypeScript
import {
|
|
type ActivityStandards,
|
|
type Levels,
|
|
} from "$lib/services/calculator/main";
|
|
import { Activity, clamp, Gender, range } from "$lib/services/calculator/util";
|
|
import allStandards from "./standards.json";
|
|
import { levenbergMarquardt as LM } from "ml-levenberg-marquardt";
|
|
|
|
const activityStandards = allStandards as ActivityStandards;
|
|
|
|
for (const activity of Object.values(Activity)) {
|
|
const DATASET_MAX_LEVEL = 5;
|
|
const NEW_LEVEL_COUNT = 2;
|
|
const RATIO_CLAMP = 1.4;
|
|
|
|
function expDecayModel([A, B, C]: number[]) {
|
|
return (i: number) => A * Math.exp(-B * i) + C;
|
|
}
|
|
|
|
for (const gender of Object.values(Gender)) {
|
|
const standardsByGender = activityStandards[activity].standards.filter(
|
|
(s) => s["metrics"].gender === gender,
|
|
);
|
|
const ages = [...new Set(standardsByGender.map((s) => s.metrics.age))];
|
|
for (const age of ages) {
|
|
const standards = standardsByGender.filter(
|
|
(s) => s.metrics.age === age,
|
|
);
|
|
|
|
const data = {
|
|
x: [] as number[],
|
|
y: [] as number[],
|
|
};
|
|
|
|
const isIncreasing =
|
|
standards[0].levels["2"] > standards[0].levels["1"];
|
|
|
|
for (const i of range(DATASET_MAX_LEVEL).slice(0, -1)) {
|
|
const level = i + 1;
|
|
const progressionRatios: number[] = [];
|
|
for (const standard of standards) {
|
|
const curr = standard.levels[level];
|
|
const next = standard.levels[level];
|
|
progressionRatios.push(
|
|
isIncreasing ? next / curr : curr / next,
|
|
);
|
|
}
|
|
|
|
const sum = progressionRatios.reduce((p, c) => p + c, 0);
|
|
const progressionRatioAvg = sum / progressionRatios.length;
|
|
|
|
data.x.push(i);
|
|
data.y.push(progressionRatioAvg);
|
|
}
|
|
|
|
const fittedParams = LM(data, expDecayModel, {
|
|
initialValues: [0.4, 0.5, 1.1],
|
|
minValues: [0.0, 0.0, 1.02],
|
|
maxValues: [1.0, 2.0, 1.25],
|
|
maxIterations: 200,
|
|
});
|
|
const getDecayRatio = (i: number) =>
|
|
clamp(
|
|
expDecayModel(fittedParams.parameterValues)(i),
|
|
1,
|
|
RATIO_CLAMP,
|
|
);
|
|
|
|
// UPDATE THE DATA
|
|
for (const standard of standards) {
|
|
const newLevels: Levels = {};
|
|
|
|
let prev = standard.levels["1"];
|
|
for (const i of range(NEW_LEVEL_COUNT)) {
|
|
const level = i + 1;
|
|
const ratio = getDecayRatio(level - NEW_LEVEL_COUNT);
|
|
prev = isIncreasing ? prev / ratio : prev * ratio;
|
|
newLevels[level] = Math.round(prev);
|
|
}
|
|
|
|
const oldLevels = standard.levels as Levels;
|
|
for (const i of range(DATASET_MAX_LEVEL)) {
|
|
const level = i + 1;
|
|
newLevels[level + NEW_LEVEL_COUNT] = oldLevels[level];
|
|
}
|
|
standard.levels = newLevels;
|
|
console.log(newLevels);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// Bun.write("./newStandards.json", JSON.stringify(allStandards, null, 2));
|