Gagan Bhatia commited on
Commit
297c713
1 Parent(s): 1ebd2ab

Create train_model.py

Browse files
Files changed (1) hide show
  1. src/models/train_model.py +7 -0
src/models/train_model.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ from .model import Summarization
2
+ from data.make_dataset import make_dataset
3
+
4
+ def train_model():
5
+ """
6
+ Train the model
7
+ """