Update card

vitaliiznak · Sep 17, 2024 · d815c8c · d815c8c
1 parent 7461f6b
commit d815c8c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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;