fixes

dvc.yaml

@@ -26,7 +26,7 @@ stages:
       - models:
           persist: true
     metrics:
-      - reports/training_metrics.txt:
+      - reports/training_metrics.csv:
           cache: false
   eval:
     cmd: python src/models/evaluate_model.py

src/models/model.py

@@ -333,7 +333,7 @@ class Summarization:
 
         WandLogger = WandbLogger(project="summarization-dagshub")
 
-        # logger = DAGsHubLogger(metrics_path='reports/training_metrics.txt')
+        # logger = DAGsHubLogger(metrics_path='reports/training_metrics.csv')
 
         early_stop_callback = (
             [

src/models/train_model.py

@@ -28,8 +28,8 @@ def train_model():
 
     model.save_model(model_dir=params['model_dir'])
 
-    df = pd.read_json(r'wandb/latest-run/files/wandb-summary.json')
-    df.to_csv(r'reports/training_metrics.txt', index=False)
+    df = pd.read_json(r'wandb/latest-run/files/wandb-summary.json',typ='series')
+    df.to_csv(r'reports/training_metrics.csv', index=False)
 
 
 if __name__ == '__main__':