Small, primitive neural network in Kotlin
A downloadable library
Hi, this is my entry to the "Machine Learning Jam" by archuser555
I implemented this today in Kotlin, my favorite programming language, and it works well :).
It is a very basic neural network.
My test is Test.kt, where I train it on the MNIST handwritten digits dataset.
With two hidden layers of size 28, it achieves 95% accuracy on the test dataset, until it then overfits.
It reaches the best performance at ~ 20 epochs.
Status | Released |
Category | Tool |
Author | Antonio Noack |
Tags | artificial-intelligence, kotlin, machine-learning, neural-network |
Download
Download
ml.zip 4 kB
Install instructions
For compiling it, I recommend Intellij Idea.
The test dataset can be downloaded from http://yann.lecun.com/exdb/mnist/, then you just have to adjust the path to the files in Test.kt, and can run the network :)
Leave a comment
Log in with itch.io to leave a comment.