Skip to content

Instantly share code, notes, and snippets.

@juliensimon

juliensimon/dgl7.py

Last active December 20, 2019 23:49
Show Gist options
  • Save juliensimon/791fea21ec401b318f9458b300e61e26 to your computer and use it in GitHub Desktop.
Save juliensimon/791fea21ec401b318f9458b300e61e26 to your computer and use it in GitHub Desktop.
Download ZIP
DGL part 7
Raw
dgl7.py
last_epoch = all_preds[epochs-1].detach().numpy()
predicted_class = np.argmax(last_epoch, axis=-1)
print(predicted_class)
[0 0 0 0 0 0 0 0 0 0 0 0 0 0 1 1 0 0 1 0 1 0 1 1 1 1 1 1 0 1 0 1 1 1]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment