Open valeman opened 4 months ago
I just pushed support for this :)
You'll need to update the package (pip install imodelsx
), then can pass hidden_layer_sizes
list for multiple layers (like sklearn's MLP). See the second cell of the updated notebook for an example.
I was wondering if this implementation allowed for multiple hidden layers that is key in getting good performance.