深度學(xué)習(xí)TOP方案解析系列視頻課
查看數(shù)據(jù)集大小
if __name__ == '__main__':
? ?TRAIN_DF = 'data/train.csv'
? ?SUBMIT_DF = 'data/sample_submission.csv'
? ?tagged = dict([(p, w) for _, p, w in pd.read_csv(TRAIN_DF).to_records()])
? ?submit = [p for _, p, _ in pd.read_csv(SUBMIT_DF).to_records()]
? ?join = list(tagged.keys()) + submit ?# 訓(xùn)練集和測(cè)試集所有圖片名字的總和 ? ?print('給定的訓(xùn)練集大小：{}'.format(len(tagged)))
? ?print('給定的測(cè)試集大?。簕}'.format(len(submit)))
? ?print('全部圖片數(shù)量：{}'.format(len(join)))
? ?print('訓(xùn)練集前5個(gè)樣本：{}'.format(list(tagged.items())[:5]))
? ?print('測(cè)試集前5個(gè)樣本：{}'.format(submit[:5]))
標(biāo)簽：
