Skip to content

KelseyJ22/CapsuleNetwork

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 

Repository files navigation

CapsuleNetwork

TensorFlow implementation of Hinton et al's capsule network concept.

Uses TensorFlow's built-in MNIST sample dataset. To run, must have Python 2 and TensorFlow []. This is about as bare-bones of an implementation as I could come up with, so there are minimal checkpoints and evaluation happens at the end of training (unfortunately means have to train each time). Top of the to-do list is to come up with a version that doesn't require this.

Run with "python rewrite_capsule_network.py"

About

Playing around with capsule networks

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages