gagan3012 commited on
Commit
5363264
·
1 Parent(s): 0ee5810

New Updates

Browse files
src/models/model.py CHANGED
@@ -2,7 +2,7 @@ import time
2
  import torch
3
  import numpy as np
4
  import pandas as pd
5
- from datasets import load_metric
6
  from transformers import (
7
  AdamW,
8
  T5ForConditionalGeneration,
@@ -302,7 +302,9 @@ class Summarization:
302
  tokenizer=self.tokenizer, model=self.model, output=outputdir
303
  )
304
 
305
- logger = MLFlowLogger(experiment_name="Summarization",tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")
 
 
306
 
307
  early_stop_callback = (
308
  [
 
2
  import torch
3
  import numpy as np
4
  import pandas as pd
5
+ from dagshub.pytorch_lightning import DAGsHubLogger
6
  from transformers import (
7
  AdamW,
8
  T5ForConditionalGeneration,
 
302
  tokenizer=self.tokenizer, model=self.model, output=outputdir
303
  )
304
 
305
+ #logger = MLFlowLogger(experiment_name="Summarization",tracking_uri="https://dagshub.com/gagan3012/summarization.mlflow")
306
+
307
+ logger = DAGsHubLogger()
308
 
309
  early_stop_callback = (
310
  [
src/models/predict_model.py CHANGED
@@ -1,5 +1,5 @@
 
1
  from .model import Summarization
2
- from .make_dataset import make_dataset
3
 
4
  def predict_model(text):
5
  """
 
1
+ from src.data.make_dataset import make_dataset
2
  from .model import Summarization
 
3
 
4
  def predict_model(text):
5
  """