Skip to content

ashigirl96/sakurai-nmf

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

60 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

NMF Neural Nets

The NMF optimizer doesn't work perfectly. But, the NMF would get well.

Results of MNIST

Now it cannot run well. Please see test_optimizers.py or mnist.py.

But now we can get accuracy 81% for only 5 iterations without ReLU.

python sakurai_nmf/examples/mnist.py --batch_size 5000 --use_bias --num_mf_iters 5 --num_bp_iters 0
Using TensorFlow backend.
NMF-optimizer
(5/5) [Train]loss 0.606, accuracy 89.420 time, 9.197 [Test]loss 0.715, accuracy 81.312

Use only Nonlinear semi-NMF.

$ python sakurai_nmf/examples/mnist.py --batch_size 5000 --use_relu --use_bias --num_mf_iters 5 --num_bp_iters 0
Using TensorFlow backend.
NMF-optimizer
(5/5) [Train]loss 0.602, accuracy 91.480 time, 10.648 [Test]loss 0.669, accuracy 87.272

Results of Fashion MNIST

we can get accuracy 74% for only 3 iterations.

$ python sakurai_nmf/examples/mnist.py --batch_size 5000 --use_bias --num_mf_iters 5 --num_bp_iters 0 --dataset fashion
Using TensorFlow backend.
NMF-optimizer
(5/5) [Train]loss 0.579, accuracy 87.920 time, 9.588 [Test]loss 0.684, accuracy 78.3846

Releases

No releases published

Packages

No packages published

Languages