- Notifications
You must be signed in to change notification settings - Fork18
shamangary/Keras-MNIST-center-loss-with-visualization
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
Fig. (left) Softmax only. (right) Softmax with center loss
- Code explanation in Chinese.http://shamangary.logdown.com/posts/6424093
- Remove the one-hot inputs for Embedding layer and replace it by single value labels.
- There are two kinds labels: single value for center loss, and one-hot vector labels for softmax term.
- Every classes are visually seperated now :)
- Step.1Change the flag of center loss inside TYY_mnist.py
isCenterloss = True#isCenterloss = False- Step.2Run the file
KERAS_BACKEND=tensorflow python TYY_mnist.py- Anaconda
- Keras
- Tensorflow
- Others: (install with anaconda)
conda install -c anaconda scikit-learn conda install -c conda-forge matplotlibAbout
An implementation for mnist center loss training and visualization
Topics
Resources
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Releases
No releases published
Packages0
No packages published

