from model.dataset import PinyinInputDataset

dataset = PinyinInputDataset('/home/songsenand/Data/corpus/CCI-Data/')
for i, line in enumerate(dataset):
    print(line['labels'])
    if i > 10:
        break