from zipfile import ZipFile import os from sklearn.model_selection import train_test_split zipfile = "animeface-character-dataset.zip" with ZipFile(zipfile) as zip: ziplist = zip.infolist() # 集計 def categorize(file_path): paths = file_path.split('/') if len(paths) != 4: return None if paths[1] != "thumb": return None if not paths[3].endswith(".png"): return None return paths[2] files = {} for f in

