[feature] added summary quality rater

This commit is contained in:
theblackcat102
2023-01-01 10:19:25 +00:00
parent 1197dccf11
commit 168e9ca6b3
8 changed files with 251 additions and 15 deletions
-4
View File
@@ -19,8 +19,6 @@
'''
from typing import Optional, Union
import glob
import json
from dataclasses import dataclass
import numpy as np
from torch.utils.data import Dataset
@@ -145,8 +143,6 @@ class HFSummary(Dataset):
elif 'post' in data['info']:
context = data['info']['post']
if context is None:
continue
if context not in self.index2summary:
self.index2summary[len(self.index2summary)] = context