Gagan Bhatia commited on
Commit
80a254f
1 Parent(s): da68c13

Update train_model.py

Browse files
Files changed (1) hide show
  1. src/models/train_model.py +0 -1
src/models/train_model.py CHANGED
@@ -17,7 +17,6 @@ def train_model():
17
  train_df = pd.read_csv("data/processed/train.csv")
18
  eval_df = pd.read_csv("data/processed/validation.csv")
19
 
20
- eval_df = eval_df.sample(frac=params['split'], replace=True, random_state=1)
21
 
22
  model = Summarization()
23
  model.from_pretrained(model_type=params['model_type'], model_name=params['model_name'])
 
17
  train_df = pd.read_csv("data/processed/train.csv")
18
  eval_df = pd.read_csv("data/processed/validation.csv")
19
 
 
20
 
21
  model = Summarization()
22
  model.from_pretrained(model_type=params['model_type'], model_name=params['model_name'])