creating csv files code example

Example: Creating csv file

import pandas as pdimport osimport torchdevice = ("cuda" if torch.cuda.is_available() else "cpu")train_df = pd.DataFrame(columns=["img_name","label"])train_df["img_name"] = os.listdir("train/")for idx, i in enumerate(os.listdir("train/")):    if "cat" in i:        train_df["label"][idx] = 0    if "dog" in i:        train_df["label"][idx] = 1train_df.to_csv (r'train_csv.csv', index = False, header=True)