Skip to content

Instantly share code, notes, and snippets.

@alstat
Created March 8, 2018 10:46
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save alstat/e13cb240ac9be2d9822fb83e4836f53a to your computer and use it in GitHub Desktop.
Save alstat/e13cb240ac9be2d9822fb83e4836f53a to your computer and use it in GitHub Desktop.
using Flux: onehotbatch
x_vec = [float.(reshape(x_train[:, :, :, i], :)) for i in 1:1000];
X = hcat(x_vec...);
Y = onehotbatch(y_train_fine[1:1000], 0:99);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment