search

JuliaAI / MLJModels.jl

Home of the MLJ model registry and tools for model queries and mode code loading
MIT License
80 stars 27 forks source link

Feature Request: neural networks from ScikitLearn #199

Open azev77 opened 4 years ago

azev77 commented 4 years ago

MLJ has most models from ScikitLearn, but I believe it's missing neural networks: https://scikit-learn.org/stable/modules/neural_networks_supervised.html

It would be really awesome to add these models to MLJ.

tlienart commented 4 years ago

Yes this was deliberate laziness from my side due to the fact that these models are not very good but I guess an argument in favour of bringing them in is that a simple MLP can be used at a final stage of a learning network which could be useful.

This would be a reasonably simple PR if you felt up to it since it's pretty much identical to existing code for sklearn models, I'd be happy to help you with it :) otherwise I may do it but probably not very soon