推荐对话的goal感觉好麻烦呐
crazyzsunny 发布于2020-04-08 16:33 浏览:333 回复:7
0
收藏

推荐对话的goal感觉好麻烦呐

收藏
点赞
0
个赞
共7条回复 最后由鱼栖树梢回复于2020-04-15 09:33
#8鱼栖树梢回复于2020-04-15 09:33:48
#4 crazyzsunny回复
训练集不能直接用的,需要处理一下

请问官方代码不能直接跑吗?还需要处理?

0
#7谁将就谁回复于2020-04-14 18:13:29
#5 crazyzsunny回复
需要处理一下,变成demo的格式

这个demo的格式是在哪里呢

0
#6试月是你的谎言回复于2020-04-13 07:19:11

文档写的不清楚也就算了,到群里问还避而不答,真是服了,其他赛道都那么多有分的,这个赛道。。。啧啧啧

0
#5crazyzsunny回复于2020-04-10 17:30:09
#3 z1r1z6回复
你好啊,这个推荐对话的训练集是不是有问题,data_generator中有一个根据'\t'来分割句子的代码段,但是数据集里都是用空格来隔断句子的,所以整个line就是一个,只有line[0],到line[1]就报错超出范围。 [代码]
展开

需要处理一下,变成demo的格式

0
#4crazyzsunny回复于2020-04-10 17:29:39

训练集不能直接用的,需要处理一下

0
#3z1r1z6回复于2020-04-09 18:14:42

你好啊,这个推荐对话的训练集是不是有问题,data_generator中有一个根据'\t'来分割句子的代码段,但是数据集里都是用空格来隔断句子的,所以整个line就是一个,只有line[0],到line[1]就报错超出范围。

def file_reader(file_path):
    utterance = list()
    label = list()
    goal_type = list()
    goal_entity = list()
    bot_flag = list()

    with open(file_path, "r") as f:
        utt = list()
        lab = list()
        gtp = list()
        get = list()
        bfl = list()
        for line in f.readlines():
            if line == "\n":
                utterance.append(utt)
                label.append(lab)
                goal_type.append(gtp)
                goal_entity.append(get)
                bot_flag.append(bfl)
                
                utt = list()
                lab = list()
                gtp = list()
                get = list()
                bfl = list()
            else:
                line = line.split("\t")
                if line[0] == "":
                    if line[2] == "再见":
                        utt.append("再见")
                    if line[2] == "音乐推荐":
                        utt.append("给你推荐一首歌吧")
                else:
                    utt.append(line[0])
                
                if line[1] == "":
                    lab.append(int(line[2]))
                    gtp.append(line[3])
                    if line[4] == "":
                        get.append(line[3])
                    else:
                        get.append(line[4])
                else:
                    lab.append(int(line[1]))
                    gtp.append(line[2])
                    if line[3] == "":
                        get.append(line[2])
                    else:
                        get.append(line[3])
                bfl.append(line[-1].replace("\n", ""))
        f.close()
    return utterance, goal_type, goal_entity, bot_flag, label
0
#2crazyzsunny回复于2020-04-08 16:34:43

感觉goal的梳理是个大工程

0
快速回复
TOP
切换版块