Skip to content

Instantly share code, notes, and snippets.

@sdcubber

sdcubber/predict.py

Last active August 9, 2018 08:39
Show Gist options
  • Save sdcubber/5d86a5f9721cf5668614d3328429a704 to your computer and use it in GitHub Desktop.
Save sdcubber/5d86a5f9721cf5668614d3328429a704 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
predict.py
test_seq = KagglePlanetSequence('./KagglePlanetMCML.csv',
'./data/train/',
im_size=IM_SIZE,
batch_size=32, mode='test')
predictions = model.predict_generator(generator=test_seq, verbose=1)
len(predictions[1]) == len(df_train) # This is True!
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment