diff --git a/src/LearningProcessVisualizer/LearningProcessVisualizer.tsx b/src/LearningProcessVisualizer/LearningProcessVisualizer.tsx
index 04fd67b..e0f8503 100644
--- a/src/LearningProcessVisualizer/LearningProcessVisualizer.tsx
+++ b/src/LearningProcessVisualizer/LearningProcessVisualizer.tsx
@@ -58,9 +58,6 @@ const LearningProcessVisualizer: Component = () => {
Learning Rate: {data.learningRate}
*/}
);
- case 'iteration':
- console.log("Rendering iteration step");
- return {/* Iteration {data.iteration} completed, Loss: {data.loss?.toFixed(4)}
; */}
default:
console.log("Unknown step:", currentPhase);
return null;
diff --git a/src/store.ts b/src/store.ts
index 043cd87..4eaee2a 100644
--- a/src/store.ts
+++ b/src/store.ts
@@ -211,6 +211,7 @@ function updateWeights() {
const result = store.trainer.updateWeights(store.trainingConfig.learningRate);
setStore('trainingStepResult', result);
+ setStore('trainingState', 'weightUpdateResults', result); // Add this line
setStore('network', store.trainer.network);
setStore('trainingState', 'currentPhase', 'update');
});
diff --git a/src/types.ts b/src/types.ts
index 06e5bad..1a9a07c 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -13,7 +13,7 @@ export type AppState = {
// Training state
trainingState: {
- currentPhase: 'idle',
+ currentPhase: 'idle'| 'forward'| 'loss' | 'backward' | 'update',
iteration: 0,
currentLoss: null,
forwardStepResults: [],
@@ -38,6 +38,8 @@ export type AppState = {
trainingRuns: TrainingRun[]; // Add this line
};
+type TrainingRun = any
+
export type BackwardStepGradients = {
neuron: number;
weights: number;