import torchaudio
import matplotlib.pyplot as plt

yesno_data = torchaudio.datasets.YESNO(root='./data', download=True)

waveform, sample_rate, labels = yesno_data[0]

print(f"Waveform shape: {waveform.shape}")
print(f"Sample rate: {sample_rate}")
print(f"Labels: {labels}")

plt.figure()
plt.plot(waveform.t().numpy())
plt.title("Waveform")
plt.xlabel("Sample")
plt.ylabel("Amplitude")
plt.show()