Spaces:
Runtime error
Runtime error
fixes
Browse files
src/models/evaluate_model.py
CHANGED
@@ -3,6 +3,7 @@ import yaml
|
|
3 |
|
4 |
from model import Summarization
|
5 |
import pandas as pd
|
|
|
6 |
|
7 |
|
8 |
def evaluate_model():
|
@@ -17,6 +18,9 @@ def evaluate_model():
|
|
17 |
model.load_model(model_type=params['model_type'], model_dir=params['model_dir'])
|
18 |
results = model.evaluate(test_df=test_df, metrics=params['metric'])
|
19 |
|
20 |
-
with
|
21 |
-
|
22 |
-
|
|
|
|
|
|
|
|
3 |
|
4 |
from model import Summarization
|
5 |
import pandas as pd
|
6 |
+
import json
|
7 |
|
8 |
|
9 |
def evaluate_model():
|
|
|
18 |
model.load_model(model_type=params['model_type'], model_dir=params['model_dir'])
|
19 |
results = model.evaluate(test_df=test_df, metrics=params['metric'])
|
20 |
|
21 |
+
with open('reports/metrics.txt', 'w') as fp:
|
22 |
+
json.dump(results, fp)
|
23 |
+
|
24 |
+
|
25 |
+
if __name__ == '__main__':
|
26 |
+
evaluate_model()
|