RustNN
RustNN copied to clipboard
Improvement and feature addition and Update from deprecated libraries to new ones
At first, I did not plan to create a pull request and to keep everything clean and abstract. But I think the improvements got really useful, so I decided to clean up and update the doc and README accordingly, but I don't know, if everything is still good enough. If there is something minor not ok, please tell me, so I can fix and you can merge safely.
Oh sorry, I forgot to add the .trayis.yml. It is in now.
This fork adds L2 regularization and several activation functions to the original crate. Additionally, there are a few minor improvements. Lambda can be set just like the learning rate. The activation functions for hidden and output gets set in NN::new as second and third parameter respectively.
@jackm321 I'd like to see this PR merged. I tried using the nn crate in a WebAssembly project but it failed to build because nn relies on the outdated rustc-serialize package which doesn't compile for WebAssembly (See https://github.com/rustwasm/wasm-bindgen/issues/1488). This change would fix that problem.