gagan3012 commited on
Commit
e334aa5
·
1 Parent(s): f50e93d
Files changed (1) hide show
  1. src/models/model.py +4 -3
src/models/model.py CHANGED
@@ -303,7 +303,8 @@ class Summarization:
303
  tokenizer=self.tokenizer, model=self.model, output=outputdir
304
  )
305
 
306
- MLlogger = MLFlowLogger(experiment_name="Summarization",tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")
 
307
 
308
  logger = DAGsHubLogger(metrics_path='reports/metrics.txt')
309
 
@@ -324,7 +325,7 @@ class Summarization:
324
  gpus = 1 if use_gpu else 0
325
 
326
  trainer = Trainer(
327
- logger=[logger,MLlogger],
328
  callbacks=early_stop_callback,
329
  max_epochs=max_epochs,
330
  gpus=gpus,
@@ -463,7 +464,7 @@ class Summarization:
463
  'Rouge_2 High recall': results["rouge2"].high.recall,
464
  'Rouge_2 High F1': results["rouge2"].high.fmeasure,
465
  },
466
- 'Rouge L':{
467
  'Rouge_L Low Precision': results["rougeL"].low.precision,
468
  'Rouge_L Low recall': results["rougeL"].low.recall,
469
  'Rouge_L Low F1': results["rougeL"].low.fmeasure,
 
303
  tokenizer=self.tokenizer, model=self.model, output=outputdir
304
  )
305
 
306
+ MLlogger = MLFlowLogger(experiment_name="Summarization",
307
+ tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")
308
 
309
  logger = DAGsHubLogger(metrics_path='reports/metrics.txt')
310
 
 
325
  gpus = 1 if use_gpu else 0
326
 
327
  trainer = Trainer(
328
+ logger=[logger, MLlogger],
329
  callbacks=early_stop_callback,
330
  max_epochs=max_epochs,
331
  gpus=gpus,
 
464
  'Rouge_2 High recall': results["rouge2"].high.recall,
465
  'Rouge_2 High F1': results["rouge2"].high.fmeasure,
466
  },
467
+ 'Rouge L': {
468
  'Rouge_L Low Precision': results["rougeL"].low.precision,
469
  'Rouge_L Low recall': results["rougeL"].low.recall,
470
  'Rouge_L Low F1': results["rougeL"].low.fmeasure,