Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
cdce7a5
1
Parent(s):
d6b264b
Update train_model.py
Browse files
src/models/train_model.py
CHANGED
@@ -1,3 +1,5 @@
|
|
|
|
|
|
1 |
from src.models.model import Summarization
|
2 |
import pandas as pd
|
3 |
|
@@ -6,6 +8,9 @@ def train_model():
|
|
6 |
"""
|
7 |
Train the model
|
8 |
"""
|
|
|
|
|
|
|
9 |
# Load the data
|
10 |
train_df = pd.read_csv('../../data/processed/train.csv')
|
11 |
eval_df = pd.read_csv('../../data/processed/validation.csv')
|
|
|
1 |
+
import yaml
|
2 |
+
|
3 |
from src.models.model import Summarization
|
4 |
import pandas as pd
|
5 |
|
|
|
8 |
"""
|
9 |
Train the model
|
10 |
"""
|
11 |
+
with open("params.yml") as f:
|
12 |
+
params = yaml.safe_load(f)
|
13 |
+
|
14 |
# Load the data
|
15 |
train_df = pd.read_csv('../../data/processed/train.csv')
|
16 |
eval_df = pd.read_csv('../../data/processed/validation.csv')
|