search

gau820827 / AI-writer_Data2Doc

PyTorch Implementation of NBA game summary generator.
Apache License 2.0
82 stars 22 forks source link

Bug while creating the Language for Summaries #48

Open rchanda opened 6 years ago

rchanda commented 6 years ago

https://github.com/gau820827/AI-writer_Data2Doc/blob/master/train/dataprepare.py

for v in data_set:
        for triplet in v.triplets:
            #  Example:
            #       triplet ('TEAM-FT_PCT', 'Cavaliers', '68')
            #               ('FGA', 'Tyler Zeller', '6')
            rt.addword(triplet[0])
            re.addword(triplet[1])
            rm.addword(triplet[2])
            summarize.addword(triplet[2])
    for v in data_set:
        for word in v.summary:
            # summary
            summarize.addword(word)

triplet[2] (r.m) shouldn't be added in the language of summarize. summarize.addword(triplet[2])

weikaipan commented 6 years ago

Hi, Thanks for pointing it out, we will fix it soon.