python生成訓練集和測試集,原數據以txt給出

源數據: 第二列爲id號; 將以txt存儲的圖片分成train和test集,train和test比例按每個id的總圖片數的7:3,實現代碼如下所示: id_a = [] #存放所有的標籤號 id_f = [] #存放圖片數大於等於20的的標籤號 id_num =[] path = '' save_train_path = '' save_val_path = '' with codecs.open
相關文章
相關標籤/搜索