# -- Code Cell --
import torchaudio.transforms as T
import torch
import torch.nn as nn
from torch.utils.data import DataLoader,Dataset
import torchaudio
import pandas as pd

# -- Code Cell --
train = pd.read_csv('train.csv')
train.head()
mapp = {}
for i in range(len(train['class'].unique().tolist())):
    mapp[train['class'].unique().tolist()[i]] = i

# -- Code Cell --
mapp

# -- Code Cell --
labels = train['class'].map(mapp)

# -- Code Cell --
labels

# -- Code Cell --
import torch, torchaudio
print(torch.__version__)
print(torchaudio.__version__)

# -- Code Cell --
for i in range(len(train)):
    waveform, sr = torchaudio.load(f"./audio/S_{i:05d}.wav")

# -- Code Cell --
