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