Commit fd9e6989 authored by T.H.C. Heshan's avatar T.H.C. Heshan

Update model.py

parent 808d3c2b
...@@ -86,7 +86,7 @@ val_spectrogram_ds = val_spectrogram_ds.cache().prefetch(tf.data.AUTOTUNE) ...@@ -86,7 +86,7 @@ val_spectrogram_ds = val_spectrogram_ds.cache().prefetch(tf.data.AUTOTUNE)
test_spectrogram_ds = test_spectrogram_ds.cache().prefetch(tf.data.AUTOTUNE) test_spectrogram_ds = test_spectrogram_ds.cache().prefetch(tf.data.AUTOTUNE)
#use a simple convolutional neural network (CNN), since you have transformed the audio files into spectrogram images #use a simple convolutional neural network (CNN), since the audio files are transformed into spectrogram images
#tf.keras.layers.Resizing: to downsample the input to enable the model to train faster. #tf.keras.layers.Resizing: to downsample the input to enable the model to train faster.
#tf.keras.layers.Normalization: to normalize each pixel in the image based on its mean and standard deviation #tf.keras.layers.Normalization: to normalize each pixel in the image based on its mean and standard deviation
input_shape = example_spectrograms.shape[1:] input_shape = example_spectrograms.shape[1:]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment