GoogleCloudPlatform / ml-design-patterns

Source code accompanying O'Reilly book: Machine Learning Design Patterns
Apache License 2.0
1.89k stars 532 forks source link

Chapter 5: Two Phase Predictions #11

Open mshearer0 opened 4 years ago

mshearer0 commented 4 years ago

NameError: name 'image_batch' is not defined feature_batch = mobilenet(image_batch)

Suggest: train_image, train_label = next(train_data_gen) feature_batch = mobilenet(train_image)

Phase 2: identifying instrument sounds, needs the '/audio_train_spectro' directory repopulated with images as the files are moved into audio_spectros/not_instrument/ or audio_spectros/instrument/ in the previous example

Similarly suggest:

feature_batch = vgg_model(image_instrument_train)