mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-03 10:06:44 +00:00
[feature] ColossalEval: Evaluation Pipeline for LLMs (#4786)
* Add ColossalEval * Delete evaluate in Chat --------- Co-authored-by: Xu Yuanchen <yuanchen.xu00@gmail.com> Co-authored-by: Tong Li <tong.li352711588@gmail.com>
This commit is contained in:
@@ -0,0 +1,8 @@
|
||||
def get_data_per_category(data, categories):
|
||||
data_per_category = {category: [] for category in categories}
|
||||
for item in data:
|
||||
category = item["category"]
|
||||
if category in categories:
|
||||
data_per_category[category].append(item)
|
||||
|
||||
return data_per_category
|
Reference in New Issue
Block a user