New models

dvc.lock

@@ -5,76 +5,76 @@ stages:
     deps:
     - path: data/processed/train.csv
       md5: 5331b9c32b2d097d8d7aca01de5524bc
-      size: 1198262
+      size: 1198550
     - path: data/processed/validation.csv
       md5: 6069153a075b00dfb6d9e0843dd2da89
-      size: 52739
+      size: 52753
     - path: model_params.yml
       md5: 1bf2edf25e851cc9cd3be75fbd9905a3
-      size: 177
+      size: 188
     - path: src/models/train_model.py
       md5: f7d1121426c3d5530c2b9697cb7ac74a
-      size: 951
+      size: 991
     outs:
     - path: models
-      md5: fc37870a93db61b94af9f0847577f09b.dir
-      size: 243476333
-      nfiles: 5
+      md5: 687b74d21b979d09bfdfff0722eae305.dir
+      size: 244268079
+      nfiles: 6
     - path: reports/training_metrics.csv
-      md5: 3b309def91a32e521acd23b163742522
+      md5: e0f69aef2d1d7f02c561f8c7801c3a20
       size: 320
   eval:
     cmd: python src/models/evaluate_model.py
     deps:
     - path: data/processed/test.csv
       md5: 3eec94ac211c76363a3d968663b82d02
-      size: 39574
+      size: 39586
     - path: model_params.yml
       md5: 1bf2edf25e851cc9cd3be75fbd9905a3
-      size: 177
+      size: 188
     - path: models
-      md5: fc37870a93db61b94af9f0847577f09b.dir
-      size: 243476333
-      nfiles: 5
+      md5: 687b74d21b979d09bfdfff0722eae305.dir
+      size: 244268079
+      nfiles: 6
     - path: src/models/evaluate_model.py
-      md5: 89edb77aaab3055605ae6db2e21eab82
-      size: 705
+      md5: ba0f4c7aa238de433cd389e9d2e66d64
+      size: 734
     outs:
     - path: reports/evaluation_metrics.csv
-      md5: eaa3bf017026aa1be31560f308fff78e
-      size: 2122
+      md5: aef31e3b826508ffe01ab4f579715ff8
+      size: 2117
   process_data:
     cmd: python src/data/process_data.py
     deps:
     - path: data/raw
       md5: 2ab20ac1b58df875a590b07d0e04eb5b.dir
-      size: 1358833013
+      size: 1359144987
       nfiles: 3
     - path: data_params.yml
       md5: a68eabf79c3b3e28afb05baa1944bbc7
-      size: 32
+      size: 33
     - path: src/data/process_data.py
-      md5: 68db554a69a0c8ce807907afa2be5e9c
-      size: 521
+      md5: 112f5e26a26956c508da053b4e2a950e
+      size: 530
     outs:
     - path: data/processed/test.csv
       md5: 3eec94ac211c76363a3d968663b82d02
-      size: 39574
+      size: 39586
     - path: data/processed/train.csv
       md5: 5331b9c32b2d097d8d7aca01de5524bc
-      size: 1198262
+      size: 1198550
     - path: data/processed/validation.csv
       md5: 6069153a075b00dfb6d9e0843dd2da89
-      size: 52739
+      size: 52753
   download_data:
     cmd: python src/data/make_dataset.py
     deps:
     - path: data_params.yml
       md5: a68eabf79c3b3e28afb05baa1944bbc7
-      size: 32
+      size: 33
     - path: src/data/make_dataset.py
       md5: a0667f4ad8c06551609bd0bf950167b7
-      size: 776
+      size: 801
     outs:
     - path: data/raw
       md5: 2ab20ac1b58df875a590b07d0e04eb5b.dir

reports/evaluation_metrics.csv

@@ -1,37 +1,37 @@
 Name,Value,Timestamp,Step
-"Rouge_1 Low Precision",0.23786550570641482,1628587253223,1
-"Rouge_1 Low recall",0.23355396379384713,1628587253223,1
-"Rouge_1 Low F1",0.23602599457077003,1628587253223,1
-"Rouge_1 Mid Precision",0.3569471852499436,1628587253223,1
-"Rouge_1 Mid recall",0.31915939075819916,1628587253223,1
-"Rouge_1 Mid F1",0.3317618573023773,1628587253223,1
-"Rouge_1 High Precision",0.4726861301480842,1628587253223,1
-"Rouge_1 High recall",0.4019654200001146,1628587253223,1
-"Rouge_1 High F1",0.4298956952594035,1628587253223,1
-"Rouge_2 Low Precision",0.06184772400193972,1628587253223,1
-"Rouge_2 Low recall",0.05626972412346313,1628587253223,1
-"Rouge_2 Low F1",0.058680298802341754,1628587253223,1
-"Rouge_2 Mid Precision",0.1367034298993256,1628587253223,1
-"Rouge_2 Mid recall",0.11953160646342464,1628587253223,1
-"Rouge_2 Mid F1",0.12485064123505887,1628587253223,1
-"Rouge_2 High Precision",0.22739029631016827,1628587253223,1
-"Rouge_2 High recall",0.18851628169809986,1628587253223,1
-"Rouge_2 High F1",0.20306657551189072,1628587253223,1
-"Rouge_L Low Precision",0.18248956154159507,1628587253223,1
-"Rouge_L Low recall",0.18048774357814204,1628587253223,1
-"Rouge_L Low F1",0.18151380309623336,1628587253223,1
-"Rouge_L Mid Precision",0.2614974838710314,1628587253223,1
-"Rouge_L Mid recall",0.24286688705755238,1628587253223,1
-"Rouge_L Mid F1",0.24674586991996245,1628587253223,1
-"Rouge_L High Precision",0.3574471638807763,1628587253223,1
-"Rouge_L High recall",0.30836083808542225,1628587253223,1
-"Rouge_L High F1",0.32385446385474176,1628587253223,1
-"rougeLsum Low Precision",0.21468633089019287,1628587253223,1
-"rougeLsum Low recall",0.2057771050364415,1628587253223,1
-"rougeLsum Low F1",0.21170611912426093,1628587253223,1
-"rougeLsum Mid Precision",0.3060593850789648,1628587253223,1
-"rougeLsum Mid recall",0.27733553744690076,1628587253223,1
-"rougeLsum Mid F1",0.28530501988436374,1628587253223,1
-"rougeLsum High Precision",0.4094614601758424,1628587253223,1
-"rougeLsum High recall",0.34640369291505535,1628587253223,1
-"rougeLsum High F1",0.36454440079714096,1628587253223,1
+"Rouge_1 Low Precision",0.23647975827999754,1629047096212,1
+"Rouge_1 Low recall",0.22831287096027664,1629047096212,1
+"Rouge_1 Low F1",0.23418947217828084,1629047096212,1
+"Rouge_1 Mid Precision",0.3560591060428998,1629047096212,1
+"Rouge_1 Mid recall",0.3127238747956226,1629047096212,1
+"Rouge_1 Mid F1",0.32913914058036986,1629047096212,1
+"Rouge_1 High Precision",0.47201228644792464,1629047096212,1
+"Rouge_1 High recall",0.3972251342872619,1629047096212,1
+"Rouge_1 High F1",0.42882258560945286,1629047096212,1
+"Rouge_2 Low Precision",0.06184772400193972,1629047096212,1
+"Rouge_2 Low recall",0.05546222700639858,1629047096212,1
+"Rouge_2 Low F1",0.05751992748351973,1629047096212,1
+"Rouge_2 Mid Precision",0.1367034298993256,1629047096212,1
+"Rouge_2 Mid recall",0.11630414726089211,1629047096212,1
+"Rouge_2 Mid F1",0.12395501575993619,1629047096212,1
+"Rouge_2 High Precision",0.22739029631016827,1629047096212,1
+"Rouge_2 High recall",0.18541872125083267,1629047096212,1
+"Rouge_2 High F1",0.2017332889523576,1629047096212,1
+"Rouge_L Low Precision",0.17657404838838545,1629047096212,1
+"Rouge_L Low recall",0.1730824748047936,1629047096212,1
+"Rouge_L Low F1",0.17073717686838885,1629047096212,1
+"Rouge_L Mid Precision",0.25587629049009775,1629047096212,1
+"Rouge_L Mid recall",0.22749719316874228,1629047096212,1
+"Rouge_L Mid F1",0.23669721679498185,1629047096212,1
+"Rouge_L High Precision",0.3527312699546978,1629047096212,1
+"Rouge_L High recall",0.2946306224842409,1629047096212,1
+"Rouge_L High F1",0.3163820193302396,1629047096212,1
+"rougeLsum Low Precision",0.2085868171757611,1629047096212,1
+"rougeLsum Low recall",0.20289459133800977,1629047096212,1
+"rougeLsum Low F1",0.20387245278320992,1629047096212,1
+"rougeLsum Mid Precision",0.30344871037135057,1629047096212,1
+"rougeLsum Mid recall",0.2675052788335701,1629047096212,1
+"rougeLsum Mid F1",0.2807970688894797,1629047096212,1
+"rougeLsum High Precision",0.4069305916918395,1629047096212,1
+"rougeLsum High recall",0.3338834625041223,1629047096212,1
+"rougeLsum High F1",0.3615219242022333,1629047096212,1

reports/training_metrics.csv

@@ -1,9 +1,9 @@
 Name,Value,Timestamp,Step
-"val_loss",2.615034580230713,1628591864766,0
-"epoch",0,1628591864766,0
-"val_loss",2.6141018867492676,1628591893945,1
-"epoch",1,1628591893945,1
-"val_loss",2.6132164001464844,1628591923101,2
-"epoch",2,1628591923101,2
-"val_loss",2.612450361251831,1628591951319,3
-"epoch",3,1628591951319,3
+"val_loss",2.6151387691497803,1629046733160,0
+"epoch",0,1629046733160,0
+"val_loss",2.614091634750366,1629046834859,1
+"epoch",1,1629046834859,1
+"val_loss",2.612950325012207,1629046914792,2
+"epoch",2,1629046914792,2
+"val_loss",2.6119742393493652,1629046993449,3
+"epoch",3,1629046993449,3