|
@@ -10,21 +10,17 @@ def jieba_tokenizer(text):
|
|
|
return list(jieba.cut(text))
|
|
|
# 定义问题模板
|
|
|
template_dict = {
|
|
|
- "8": [
|
|
|
- "某年省间交易电量按交易周期划分的电量是多少?",
|
|
|
- "某年省间交易电量按交易类型划分的电量是多少?",
|
|
|
- "某年省间交易电量按发电类型划分的电量是多少?",
|
|
|
- "某年省间交易电量按交易方式划分的电量是多少?",
|
|
|
- ],
|
|
|
- "19": [
|
|
|
- "省间交易正在组织的交易有多少?",
|
|
|
- "省间交易当月完成的交易有多少?",
|
|
|
- "省间交易当年完成的交易有多少?",
|
|
|
- "省间交易当年达成的电量有多少?",
|
|
|
- "省间交易当年参与交易的家次有多少?",
|
|
|
- ],
|
|
|
"1": ["某年全年累计省间交易电量是多少?"],
|
|
|
"2": ["某年某月交易电量是多少?"],
|
|
|
+ "8.1": ["某年省间交易电量按交易周期划分的电量是多少?"],
|
|
|
+ "8.2": ["某年省间交易电量按交易类型划分的电量是多少?",],
|
|
|
+ "8.3": ["某年省间交易电量按发电类型划分的电量是多少?",],
|
|
|
+ "8.4": ["某年省间交易电量按交易方式划分的电量是多少?",],
|
|
|
+ "19": ["省间交易正在组织的交易有多少?",],
|
|
|
+ "20": ["省间交易当月完成的交易有多少?",],
|
|
|
+ "21": ["省间交易当年完成的交易有多少?",],
|
|
|
+ "22": ["省间交易当年达成的电量有多少?",],
|
|
|
+ "23": ["省间交易当年参与交易的家次有多少?",],
|
|
|
}
|
|
|
# 将地点映射成相应的代码
|
|
|
def map_location_to_unit(location: str) -> str:
|