From d815c8c046160525746e9ac45d063629313d7516 Mon Sep 17 00:00:00 2001
From: Viitalii Znak <vitaliiznak@users.noreply.github.com>
Date: Tue, 17 Sep 2024 17:28:27 +0200
Subject: [PATCH] Update card

---
 src/LearningProcessVisualizer/LearningProcessVisualizer.tsx | 3 ---
 src/store.ts                                                | 1 +
 src/types.ts                                                | 4 +++-
 3 files changed, 4 insertions(+), 4 deletions(-)

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 = () => {
               <p>Learning Rate: {data.learningRate}</p> */}
             </div>
           );
-        case 'iteration':
-          console.log("Rendering iteration step");
-          return {/* <div>Iteration {data.iteration} completed, Loss: {data.loss?.toFixed(4)}</div>; */}
         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;