mirror of
https://github.com/tencentmusic/supersonic.git
synced 2025-12-20 06:34:55 +00:00
(improvement) revise evaluation and fix null pointer (#715)
This commit is contained in:
mainmain
@@ -103,6 +103,7 @@ public class BenchMarkDemoDataLoader {
|
||||
dimension1.setTypeParams(new DimensionTimeTypeParams());
|
||||
dimensions.add(dimension1);
|
||||
dimensions.add(new Dim("活跃区域", "most_popular_in", DimensionType.categorical.name(), 1));
|
||||
dimensions.add(new Dim("音乐类型名称", "g_name", DimensionType.categorical.name(), 1));
|
||||
modelDetail.setDimensions(dimensions);
|
||||
|
||||
List<Identify> identifiers = new ArrayList<>();
|
||||
@@ -129,6 +130,7 @@ public class BenchMarkDemoDataLoader {
|
||||
modelReq.setDatabaseId(1L);
|
||||
ModelDetail modelDetail = new ModelDetail();
|
||||
List<Dim> dimensions = new ArrayList<>();
|
||||
dimensions.add(new Dim("艺术家名称", "artist_name", DimensionType.categorical.name(), 1));
|
||||
dimensions.add(new Dim("国籍", "country", DimensionType.categorical.name(), 1));
|
||||
dimensions.add(new Dim("性别", "gender", DimensionType.categorical.name(), 1));
|
||||
modelDetail.setDimensions(dimensions);
|
||||
@@ -157,6 +159,7 @@ public class BenchMarkDemoDataLoader {
|
||||
List<Dim> dimensions = new ArrayList<>();
|
||||
dimensions.add(new Dim("持续时间", "duration", DimensionType.categorical.name(), 1));
|
||||
dimensions.add(new Dim("文件格式", "formats", DimensionType.categorical.name(), 1));
|
||||
dimensions.add(new Dim("艺术家名称", "artist_name", DimensionType.categorical.name(), 1));
|
||||
modelDetail.setDimensions(dimensions);
|
||||
|
||||
List<Identify> identifiers = new ArrayList<>();
|
||||
@@ -184,6 +187,7 @@ public class BenchMarkDemoDataLoader {
|
||||
Dim dimension1 = new Dim("", "imp_date", DimensionType.time.name(), 0);
|
||||
dimension1.setTypeParams(new DimensionTimeTypeParams());
|
||||
dimensions.add(dimension1);
|
||||
dimensions.add(new Dim("歌曲名称", "song_name", DimensionType.categorical.name(), 1));
|
||||
dimensions.add(new Dim("国家", "country", DimensionType.categorical.name(), 1));
|
||||
dimensions.add(new Dim("语种", "languages", DimensionType.categorical.name(), 1));
|
||||
dimensions.add(new Dim("发行时间", "releasedate", DimensionType.categorical.name(), 1));
|
||||
|
||||
@@ -74,7 +74,7 @@ public class ChatDemoLoader implements CommandLineRunner {
|
||||
addAgent1();
|
||||
addAgent2();
|
||||
addAgent3();
|
||||
addAgent4();
|
||||
//addAgent4();
|
||||
addSampleChats();
|
||||
addSampleChats2();
|
||||
updateQueryScore(1);
|
||||
@@ -248,6 +248,7 @@ public class ChatDemoLoader implements CommandLineRunner {
|
||||
}
|
||||
|
||||
agent.setAgentConfig(JSONObject.toJSONString(agentConfig));
|
||||
log.info("agent:{}", JsonUtil.toString(agent));
|
||||
agentService.createAgent(agent, User.getFakeUser());
|
||||
}
|
||||
|
||||
|
||||
@@ -39,7 +39,7 @@ public class HeadlessDemoLoader implements CommandLineRunner {
|
||||
}
|
||||
modelDataDemoLoader.doRun();
|
||||
benchMarkDemoLoader.doRun();
|
||||
duSQLDemoDataLoader.doRun();
|
||||
//duSQLDemoDataLoader.doRun();
|
||||
isLoad = true;
|
||||
}
|
||||
|
||||
|
||||
@@ -1,8 +0,0 @@
|
||||
阿里云 _10_20 5
|
||||
天猫 _10_20 5
|
||||
腾讯游戏 _10_20 5
|
||||
度小满 _10_20 5
|
||||
京东金融 _10_20 5
|
||||
|
||||
|
||||
|
||||
@@ -1,8 +0,0 @@
|
||||
张勇 _10_22 5
|
||||
马化腾 _10_22 5
|
||||
朱光 _10_22 5
|
||||
刘强东 _10_22 5
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
美国 _5_8 1
|
||||
加拿大 _5_8 1
|
||||
锡尔赫特、吉大港、库斯蒂亚 _5_8 1
|
||||
孟加拉国 _5_8 3
|
||||
美国 _3_8 1
|
||||
加拿大 _3_8 1
|
||||
锡尔赫特、吉大港、库斯蒂亚 _3_8 1
|
||||
孟加拉国 _3_8 3
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
现代 _5_9 1
|
||||
tagore _5_9 1
|
||||
蓝调 _5_9 1
|
||||
流行 _5_9 1
|
||||
民间 _5_9 1
|
||||
nazrul _5_9 1
|
||||
现代 _3_9 1
|
||||
tagore _3_9 1
|
||||
蓝调 _3_9 1
|
||||
流行 _3_9 1
|
||||
民间 _3_9 1
|
||||
nazrul _3_9 1
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
美国 _6_10 1
|
||||
印度 _6_10 2
|
||||
英国 _6_10 1
|
||||
孟加拉国 _6_10 2
|
||||
美国 _3_11 1
|
||||
印度 _3_11 2
|
||||
英国 _3_11 1
|
||||
孟加拉国 _3_11 2
|
||||
|
||||
@@ -1,2 +1,2 @@
|
||||
男性 _6_11 3
|
||||
女性 _6_11 3
|
||||
男性 _3_12 3
|
||||
女性 _3_12 3
|
||||
|
||||
@@ -1,2 +1,2 @@
|
||||
mp4 _7_14 4
|
||||
mp3 _7_14 2
|
||||
mp4 _3_14 4
|
||||
mp3 _3_14 2
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
美国 _8_16 1
|
||||
印度 _8_16 2
|
||||
英国 _8_16 1
|
||||
孟加拉国 _8_16 2
|
||||
美国 _3_17 1
|
||||
印度 _3_17 2
|
||||
英国 _3_17 1
|
||||
孟加拉国 _3_17 2
|
||||
|
||||
@@ -1,2 +1,2 @@
|
||||
英文 _8_17 2
|
||||
孟加拉语 _8_17 4
|
||||
英文 _3_18 2
|
||||
孟加拉语 _3_18 4
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
阿米·奥帕尔·霍伊 _8_19 1
|
||||
我的爱 _8_19 1
|
||||
打败它 _8_19 1
|
||||
阿杰伊阿卡什 _8_19 1
|
||||
Tumi#长袍#尼罗布 _8_19 1
|
||||
舒克诺#帕塔尔#努普尔#帕埃 _8_19 1
|
||||
阿米·奥帕尔·霍伊 _3_16 1
|
||||
我的爱 _3_16 1
|
||||
打败它 _3_16 1
|
||||
阿杰伊阿卡什 _3_16 1
|
||||
Tumi#长袍#尼罗布 _3_16 1
|
||||
舒克诺#帕塔尔#努普尔#帕埃 _3_16 1
|
||||
|
||||
@@ -1,5 +0,0 @@
|
||||
百度集团 _9_15 5
|
||||
阿里巴巴集团 _9_15 5
|
||||
深圳市腾讯计算机系统有限公司 _9_15 5
|
||||
北京京东世纪贸易有限公司 _9_15 5
|
||||
网易公司 _9_15 5
|
||||
@@ -1,4 +0,0 @@
|
||||
北京 _9_16 5
|
||||
杭州 _9_16 5
|
||||
深圳 _9_16 5
|
||||
|
||||
@@ -1,7 +0,0 @@
|
||||
李彦宏 _9_18 5
|
||||
马云 _9_18 5
|
||||
马化腾 _9_18 5
|
||||
刘强东 _9_18 5
|
||||
丁磊 _9_18 5
|
||||
|
||||
|
||||
@@ -1,7 +0,0 @@
|
||||
李彦宏 _9_19 5
|
||||
张勇 _9_19 5
|
||||
刘炽平 _9_19 5
|
||||
刘强东 _9_19 5
|
||||
丁磊 _9_19 5
|
||||
|
||||
|
||||
@@ -1,2 +1,2 @@
|
||||
root=.
|
||||
CustomDictionaryPath=data/dictionary/custom/DimValue_1_1.txt;data/dictionary/custom/DimValue_1_2.txt;data/dictionary/custom/DimValue_9_15.txt;data/dictionary/custom/DimValue_9_16.txt;data/dictionary/custom/DimValue_9_18.txt;data/dictionary/custom/DimValue_9_19.txt;data/dictionary/custom/DimValue_10_20.txt;data/dictionary/custom/DimValue_10_22.txt;
|
||||
CustomDictionaryPath=data/dictionary/custom/DimValue_1_1.txt;data/dictionary/custom/DimValue_1_2.txt;
|
||||
|
||||
Reference in New Issue
Block a user