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

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
- train_df = train_df.sample(frac=params['split'], replace=True, random_state=1)
21
  eval_df = eval_df.sample(frac=params['split'], replace=True, random_state=1)
22
 
23
  model = Summarization()
 
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()